Using pretrained models weights

Hello and than ks again for taking the time to help me @kushaj, i really appreciate it.
I tried what you suggested and it still doesn’t work :confused: . Here’s a screenshot of the code followed by the error:



I also tried to load the weights before appending the head to the model, but get the same error