Open chenpeng68 opened 3 years ago
Sorry, I didn't train my model on VOC.
@yjh0410 您好,使用voc数据集,训练的时候报错:在train.py total_loss.backward()这一行,expected type torch.cuda.FloatTensor but got torch.cuda.LongTensor,我改为total_loss.float().backward()也是一样的错误,请问大佬怎么回事呢?
hi,Excuse me,do you have training weight for resnet101 voc dataset? please send a link