Davi Neves

Results 2 issues of Davi Neves

When I test the model with the torchinfo library, i.e. by creating some input with a ```size = (1, 3, 16, 224, 224)``` to simulating my K400 dataset entry and...

I'm using the fine_tunning model in my code with this approach to load weights in torch2.6, in "init" for [ViT class](https://github.com/wgcban/adamae/blob/main/finetune_class.py): ```py [...] self.head.weight.data.mul_(init_scale) # type: ignore self.head.bias.data.mul_(init_scale) # type:...