首页
编程
java
php
前端
首页
编程
java
php
前端
pytorch torch.jit.save
torch.jit.save() 保存模型 保存地址中存在中文,出现错误,
如何实现torch.jit.save() 保存地址可以为中文??
你看下这篇博客吧, 应该有用👉 :
PyTorch模型保存torch.save()与加载torch.load()
你还可以看下pytorch参考手册中的
pytorch save_to_buffer() (torch.jit.ScriptFunction method)
除此之外, 这篇博客:
PyTorch模型保存torch.save()与加载torch.load()
中的
2)保存部分
部分也许能够解决你的问题, 你可以仔细阅读以下内容或跳转源博客中阅读:
只保存模型训练的参数(不包括网络结构)
torch
.
save
(
model
.
state_dict
(
)
,
path
)
点击展开全文