ASFF
ASFF copied to clipboard
type(self).__name__, name)) AttributeError: 'YOLOv3' object has no attribute 'module'
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'))
@huangzicheng 保存模型的问题可以这样解决,但是这里的问题怎么解决model.module.module_list[Drop_layer[i]].reset(block_size[i], keep_p[i])
@huangzicheng 保存模型的问题可以这样解决,但是这里的问题怎么解决model.module.module_list[Drop_layer[i]].reset(block_size[i], keep_p[i])
能问下 你们用哪个版本的yolo?