I’m currently using a ViT and I wanted to investigate the hidden states after fine tuning a pre-trained model. I have got hold of the values, thats fine. The problem is I don’t know precisely the format. I’ve been looking through the Docs, and they say this:
Now I cannot find where this “returned tensors” section is. I have checked the docs for other models as all models seem to follow the same implementation, BERT has the same documentation almost exactly with the same problem. The link to the source code is dead, I hunted through the github and find the ViT source code but once again it just says:
Am I being daft or is there a hole in the documentation currently?