代码如下,只需要把模型文件的路径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 版权协议，转载请附上原文出处链接和本声明。