自己记录用的:
torch.save(net.module.state_dict(), os.path.join(args.save, 'model.pth'))
改为
torch.save(net.state_dict(), os.path.join(args.save, 'model.pth'))
自己记录用的:
torch.save(net.module.state_dict(), os.path.join(args.save, 'model.pth'))
改为
torch.save(net.state_dict(), os.path.join(args.save, 'model.pth'))