博客

如何加载把深度学习保存的模型.pth或者.module打印出来,保存在txt文件中

  • Post author:
  • Post category:其他


代码如下,只需要把模型文件的路径pthfile更改一下即可. 

import torch

pthfile = r'/home/bob/jj/pysot/snapshot_/checkpoint_e27.pth'
net = torch.load(pthfile, map_location='cpu')
net = dict(net)

with open('test.txt', 'a') as file0:
    print(net, file=file0)

这样模型信息就都保存在test.txt文件中了.

(可以点个赞不, ^ _ ^ )


版权声明：本文为weixin_44493244原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。