cpu上训练的模型放到gpu上跑:
torch.load(‘model1.pth’, map_location=lambda storage, loc: storage.cuda(0))
cpu上训练的模型放到gpu上继续训练
猜你喜欢
转载自blog.csdn.net/qq_43663979/article/details/129399832
今日推荐
周排行