Pytorch--训练并保存的模型利用 torch.load() 在GPU、CPU上加载
# 直接加载模型model.load_state_dict(torch.load('./data/my_model.pkl'))#GPU训练的模型加载到CPU上:model.load_state_dict(torch.load('./data/my_model.pkl', map_location=lambda storage, loc: storage))#加载到GPU1上:model.load_state_dict(torch.load('./data/my_model.pkl',...
原创
2020-10-05 19:43:15 ·
6859 阅读 ·
0 评论