PyTorch 快速查看 pth 文件保存的参数
不需要 run 什么脚本来 load 然后 print,直接用命令行解决:(注:假设该环境下有 torch 包,该文件名 t.pth)import torch# 命令行是逐行立即执行的content = torch.load('t.pth')print(content.keys())# keys()# 之后有其他需求比如要看 key 为 model 的内容有啥print(content['mode
文章共173字 · 阅读需要大约1分钟
一键AI生成摘要,助你高效阅读
问答
·
不需要 run 什么脚本来 load 然后 print,直接用命令行解决:
(注:假设该环境下有 torch 包,该文件名 t.pth
)
import torch # 命令行是逐行立即执行的
content = torch.load('t.pth')
print(content.keys()) # keys()
# 之后有其他需求比如要看 key 为 model 的内容有啥
print(content['model'])
进行快速检查非常方便
更多推荐
已为社区贡献1条内容
所有评论(0)