cpu上训练的模型放到gpu上跑:
torch.load(‘model1.pth’, map_location=lambda storage, loc: storage.cuda(0))
cpu上训练的模型放到gpu上继续训练
最新推荐文章于 2024-04-17 14:39:51 发布
cpu上训练的模型放到gpu上跑:
torch.load(‘model1.pth’, map_location=lambda storage, loc: storage.cuda(0))