我如何知道 Pytorch 中预训练模型的架构？

Posted 2023-03-12

【中文标题】我如何知道 Pytorch 中预训练模型的架构？

【英文标题】：How can i know the architecture of pre-trained model in Pytorch?

【发布时间】：2020-12-08 11:30:45

【问题描述】：

我已经下载了这个用于人脸识别的预训练模型“model_ir_se50.pth”。它给出了非常好的结果。我怎么知道它的架构？

【问题讨论】：

这能回答你的问题吗？ Model summary in pytorch

【参考方案1】：

简单的print(net) #replace net with the network variable 将根据您的需要工作。

【讨论】：

你的意思是这样的：import torch pretrained = torch.load("model_ir_se50.pth") print(pretrained.net())

print(pretrained)

它的输出非常大..但是你可以从我正在粘贴的一部分中得到一个想法。 OrderedDict([('input_layer.0.weight', tensor([[[[ 5.0732e-02, 6.5002e-03, 1.9098e-02],....

你能编辑你的帖子并添加你用来获取结果的代码吗？