gpu -> cpu
原始的
torch.load('modelparameters.pth')
CPU运行
torch.load('modelparameters.pth', map_location=lambda storage, loc: storage)
主要是记录一下自己平时遇到的问题,和大家分享一下
如有侵犯,请联系我
点个赞再走呗
gpu -> cpu
原始的
torch.load('modelparameters.pth')
CPU运行
torch.load('modelparameters.pth', map_location=lambda storage, loc: storage)
主要是记录一下自己平时遇到的问题,和大家分享一下
如有侵犯,请联系我
点个赞再走呗