pth文件网络,结构可视化

import torch
net=torch.load('C:/Users/yan/.cache/torch/hub/checkpoints/swin_base_patch4_window7_224_22kto1k.pth')
print(net)
print(type(net))
print(len(net))
for k in net.keys():
  print(k)
for key,value in net["model"].items():
  print(key,value.size(),sep=" ")

你可能感兴趣的:(c++,算法前期摸索,4个月,深度学习攻略,深度学习,python,pytorch)