Loading pre-trained weights properly in Pytorch
Loading pre-trained weights properly in Pytorch Question: I would like to perform transfer learning by loading a pretrained vision transformer model, modify its last layer and training it with my own data. Hence, I am loading my dataset perform the typical transformation similar to the ImageNet, then, load the model, disable the grad from all …