我们的工作现在要部署到服务器上 ,服务器上面没有GPU,所以要将GPU改为CPU处理,于是训练的模型要动一动,在load的时候加上map_location='cpu'
gpu->cpu
model = torch.load(‘xxx.pkl’, map_location=‘cpu’)
CPU->GPU1
torch.load(‘gen_500000.pkl’, map_location=lambda storage, loc: storage.cuda(1))
我们的工作现在要部署到服务器上 ,服务器上面没有GPU,所以要将GPU改为CPU处理,于是训练的模型要动一动,在load的时候加上map_location='cpu'
gpu->cpu
model = torch.load(‘xxx.pkl’, map_location=‘cpu’)
CPU->GPU1
torch.load(‘gen_500000.pkl’, map_location=lambda storage, loc: storage.cuda(1))