pytorch 查看网络模型字典Key

最新推荐文章于 2023-03-02 12:07:51 发布

夏日清风有你

最新推荐文章于 2023-03-02 12:07:51 发布

阅读量972

点赞数

分类专栏： PyTorch 文章标签： pytorch 深度学习 python

本文链接：https://blog.csdn.net/fanlily913/article/details/125172490

版权

PyTorch 专栏收录该内容

50 篇文章 1 订阅

订阅专栏

import os
os.environ["KMP_DUPLICATE_LIB_OK"] = "TRUE"
import torch
chpt = torch.load('./best_model.pth')
# chpt = torch.load('./VGG16A_BCE.pth')
# epoch = chpt['epoch']
# loss  = chpt['loss']

for k, v in chpt.items():
  print(k)

for k, v in chpt["model_state_dict"].items():
  print(k)

# for k, v in chpt["optimizer_state_dict"].items():
#   print(k)