模型参数大小的问题 about classification-pytorch HOT 14 CLOSED

有没有一种可能，pth是存储空间？

from classification-pytorch.

from classification-pytorch.

summary.py

from classification-pytorch.

summary.py

我试了一下，其他都可以出来，但是vit会报如下的错(256x256时候会有，但224的时候不会报错)，是什么原因呀？
但我在train的时候可以用256x256的大小train vit

from classification-pytorch.

vit也有个input shape要设置

from classification-pytorch.

vit也有个input shape要设置

好的，那在train的时候不需要专门到vit里面设置是因为train里面的input_shape=[x,x]会覆盖vit中的吗？

还有就是我得出的结果和ResNet原文中不一样，是因为图像的输入大小和num_classes不一样吗？

from classification-pytorch.

1、对的
2、差距应该不大吧？结构绝对是对的。

from classification-pytorch.

差距还挺大的，ResNet50 跑出来是86M，文章里写的25.6M
我的大小是(3,256,256)

from classification-pytorch.

你不知道哪个是参数量？

from classification-pytorch.

你不知道哪个是参数量？

啊？我知道吧…… 就是下面的Params size

图上是vit的结果；然后我跑resnet50的时候是86M

from classification-pytorch.

是total params…

from classification-pytorch.

是total params…
嗷嗷好的 ,那就是用这里的total params/1024/1024 得到最后的MB ，是吗？

from classification-pytorch.

直接/1e6应该，不用1024好像。

from classification-pytorch.

直接/1e6应该，不用1024好像。

好的谢谢，了解了👌

from classification-pytorch.