后来发现是因为使用
torch.save(model,'save.pt')
会保存整个文件时会默认保存训练py文件的父目录,用torch.load导入文件时搜索路径必须有此父路径,否则将会提示no model named model这样的错误
解决办法是使用sys.path.apend把该父目录加入搜索路径中
sys.path.append(r'训练py文件的父目录')
若使用
torch.save(model.state_dict(), 'save.pt')
可能就没有这样的问题了