2018年7月14日
pytorch将cpu训练好的模型参数load到gpu上，或者gpu->cpu上
摘要： 假设我们只保存了模型的参数(model.state_dict())到文件名为modelparameters.pth, model = Net() 1. cpu -> cpu或者gpu -> gpu: checkpoint = torch.load('modelparameters.pth') mod 阅读全文
posted @ 2018-07-14 23:47 琴影 阅读(9726) 评论(0) 推荐(0) 编辑