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 检测结果:

 RefineDet -pytorch代码记录_第1张图片

 

3 vgg预训练 loss变为了nan

RefineDet -pytorch代码记录_第2张图片

 

转载于:https://www.cnblogs.com/SuckChen/p/10204068.html

你可能感兴趣的:(人工智能)