栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

pth文件读取

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

pth文件读取

pth文件读取
参考知乎:https://zhuanlan.zhihu.com/p/84797438pytorch中保存的模型文件.pth深入解析

# -*- coding: utf-8 -*-
import torch
net = torch.load('iter_100000.pth')#加载模型

print(type(net))#
print(len(net))#3

#for k in net.keys():
    # print(k)##meta   state_dict  optimizer

#meta   state_dict  optimizer这三项逐一加载查看
#print(net['meta'])

# for key,value in net['meta'].items():
    # print(key,value)


# for key,value in net['state_dict'].items():
    # print(key,value)

for key,value in net['optimizer'].items():
    print(key,value)

想知道里面到底都写了些啥?
有大佬知道嘛??

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/767216.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号