This is a bug and 'ResNet' object has no attribute 'module'

Open Johnson-yue opened this issue 5 years ago • 1 comments

https://github.com/yhenon/pytorch-retinanet/blob/a76e56eb537be476abc914587ae92ab542e2b3e4/train.py#L110

Can you provide a finetune code ??

May 07 '20 14:05 Johnson-yue

if torch.cuda.is_available(): retinanet = torch.nn.DataParallel(retinanet).cuda() so that retinanet will have attribute 'module'

Oct 10 '20 08:10 chrisway613