search

GOATmessi8 / ASFF

yolov3 with mobilenet v2 and ASFF
GNU General Public License v3.0
1.05k stars 216 forks source link

type(self).__name__, name)) AttributeError: 'YOLOv3' object has no attribute 'module' #70

Open huangzicheng opened 4 years ago

huangzicheng commented 4 years ago

I solve this problem , change it with:

model_dict = dict({ 'epoch': epoch + 1, 'state_dict': model.state_dict(), })

                torch.save(model_dict, os.path.join(args.save_dir,
                        save_prefix+'_'+repr(epoch)+'.pth'))
xaohuicai commented 4 years ago

@huangzicheng 保存模型的问题可以这样解决，但是这里的问题怎么解决model.module.module_list[Drop_layer[i]].reset(block_size[i], keep_p[i])

joe660 commented 3 years ago

@huangzicheng 保存模型的问题可以这样解决，但是这里的问题怎么解决model.module.module_list[Drop_layer[i]].reset(block_size[i], keep_p[i])

能问下 你们用哪个版本的yolo？