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

[huangzicheng]

No description provided.

[huangzicheng]

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]

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

[joe660]

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

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