if args.load:
f_model = torch.load(args.load, map_location=torch.device('cpu'))
net.load_state_dict({k.replace('module.',''):v for k,v in f_model.items()})
扫一扫
评论
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
查看更多评论
添加红包
