cpu上训练的模型放到gpu上继续训练
cpu上训练的模型放到gpu上跑
·
cpu上训练的模型放到gpu上跑：
torch.load(‘model1.pth’, map_location=lambda storage, loc: storage.cuda(0))
分享最新的 NVIDIA AI Software 资源以及活动/会议信息，精选收录AI相关技术内容，欢迎大家加入社区并参与讨论。
更多推荐
0
0- 0
已为社区贡献1条内容
已为社区贡献1条内容
所有评论(0)