vit-pytorch
vit-pytorch copied to clipboard
Removing FC layer at the top of the transformer model
Hi all, I would like to ask you how I can remove the fully connected layer at the top of the transformer layer. I want to get the output of the transformer networks and not the final prediction. Moreover, I would like to ask if these models come with the pretrained weights.
Thank you.
I have same question on this, do we have any updates !?
maybe this post can help you.