lucidrains / vit-pytorch

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Removing FC layer at the top of the transformer model

loukasilias opened this issue · 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.

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

maybe this post can help you.