我是靠谱客的博主 包容世界,最近开发中收集的这篇文章主要介绍PyTorch 使用小结(更新),觉得挺不错的,现在分享给大家,希望可以做个参考。

概述

不定时更新

一、导入模型
训练好的模型直接使用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 使用小结(更新)所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错,欢迎将靠谱客网站推荐给程序员好友。

本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
点赞(53)

评论列表共有 0 条评论

立即
投稿
返回
顶部