vit-pytorch icon indicating copy to clipboard operation
vit-pytorch copied to clipboard

Removing FC layer at the top of the transformer model

Open loukasilias opened this issue 4 years ago • 2 comments

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.

loukasilias avatar Oct 09 '21 11:10 loukasilias

I have same question on this, do we have any updates !?

mghayour avatar Oct 16 '22 16:10 mghayour

maybe this post can help you.

mcihadarslanoglu avatar Feb 13 '23 15:02 mcihadarslanoglu