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
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
转载于:https://www.cnblogs.com/SuckChen/p/10204068.html