yjh0410 / CenterNet-plus

A Simple Baseline for Object Detection
55 stars 11 forks source link

pascal voc resnet101 training weight #3

Open chenpeng68 opened 3 years ago

chenpeng68 commented 3 years ago

hi,Excuse me,do you have training weight for resnet101 voc dataset? please send a link

yjh0410 commented 3 years ago

Sorry, I didn't train my model on VOC.

sunshehai commented 2 years ago

@yjh0410 您好,使用voc数据集,训练的时候报错:在train.py total_loss.backward()这一行,expected type torch.cuda.FloatTensor but got torch.cuda.LongTensor,我改为total_loss.float().backward()也是一样的错误,请问大佬怎么回事呢?