(1)pytorch在服务器上训练好的模型,下载到本地后,需要转换到CPU后才能使用。
torch.load(‘C:\Users\atr\Desktop\save\ceshi.pth’,map_location=‘cpu’)
该条语句使用前,要先预先加载模型
(2)‘model’ object has no attribute 'copy’加载模型报错
安装pytorch
容错管理库
最新推荐文章于 2022-12-12 15:41:54 发布
(1)pytorch在服务器上训练好的模型,下载到本地后,需要转换到CPU后才能使用。
torch.load(‘C:\Users\atr\Desktop\save\ceshi.pth’,map_location=‘cpu’)
该条语句使用前,要先预先加载模型
(2)‘model’ object has no attribute 'copy’加载模型报错
安装pytorch