不定时更新
一、导入模型
训练好的模型直接使用torch.save(model, path)方式保存，在其他机器或者路径进行导入时，直接使用torch.load(path)一般会报错，找不到model定义，解决方法是：
import sys
import torch
sys.path.append('模型定义.py文件所在路径')
path = '.pt文件路径'
# 这样导入不会报错
model = torch.load(path)https://github.com/pytorch/pytorch/issues/3678
最后
以上就是包容世界最近收集整理的关于PyTorch 使用小结(更新)的全部内容，更多相关PyTorch内容请搜索靠谱客的其他文章。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
发表评论 取消回复