Dilemma的博客

原创 PyTorch 使用小结(更新)

不定时更新 一、导入模型 训练好的模型直接使用torch.save(model, path)方式保存，在其他机器或者路径进行导入时，直接使用torch.load(path)一般会报错，找不到model定义，解决方法是： import sys import torch sys.path.append('模型定义.py文件所在路径') path = '.pt文件路径' # 这样导入不...