PyTorch 快速查看 pth 文件保存的参数

不需要 run 什么脚本来 load 然后 print,直接用命令行解决:

(注:假设该环境下有 torch 包,该文件名 t.pth

import torch  # 命令行是逐行立即执行的
content = torch.load('t.pth')
print(content.keys())   # keys()
# 之后有其他需求比如要看 key 为 model 的内容有啥
print(content['model'])

进行快速检查非常方便

你可能感兴趣的:(PyTorch,linux,python)