有没有简单的方法来查看训练模型的最大特征?

我正在玩预先训练的模型。
我知道一个像alexnet这样的网络在分类之前有4096个功能。在pytorch中有没有办法看到使用相同代码的所有不同预培训网络的功能数量?例如,我正在看resnet152,并且不能从阅读代码中了解到它有多少功能:https://github.com/pytorch/vision/blob/master/torchvision/models/resnet.py 1
已邀请:
匿名用户

匿名用户

赞同来自:

你可以使用
print(你的model)
就会显示每个层的详细信息。

要回复问题请先登录注册