PyTorch使用cpu加载模型运算方式

PyTorch使用使用cpu加载模型运算方式加载模型运算方式

今天小编就为大家分享一篇PyTorch使用cpu加载模型运算方式，具有很好的参考价值，希望对大家有所帮助。

一起跟随小编过来看看吧

没gpu没cuda支持的时候加载模型到cpu上计算

将将

model = torch.load(path, map_location=lambda storage, loc: storage.cuda(device))

改为改为

model = torch.load(path, map_location='cpu')