I want to print the sizes of all the layers of a pretrained model. I uae this pretrained model as self.feature in my class. The print of this pretrained model is as follows: TimeSformer( (model): VisionTransformer( (dropout): Dropout(p=0.0, inplace=False) (patch_embed): PatchEmbed( (proj): Con…

You can iterate through your parameters groups to get this info. model = YourModelClass(*args, **kwargs) for name, param in model.named_parameters(): print(name, param.shape)

Printing the dimensions of all the layers of a pretrained model

thecho7 (Suho Cho) July 26, 2022, 11:25am 2

The bellowed post is similar to your question.