RefineDet -pytorch代码记录

1、RuntimeError: copy_if failed to synchronize: device-side assert triggered

百度搜索说是标签要从0到N-1;N是类别数

 很奇怪原本没有-1,输出label_idx就是从0开始的,

 

 -1是背景类,置为0,;非背景类置为1:

 

2 无使用预训练的VGG 检测结果:

 

 

3 vgg预训练 loss变为了nan

 

posted @ 2019-01-01 02:10  SuckChen  阅读(2588)  评论(0编辑  收藏  举报