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=" ")
pth文件网络,结构可视化
最新推荐文章于 2024-04-03 19:52:50 发布