5fa1a76
1
2
3
We will actually write two: one that extracts the hidden features from a batch of images (like [BertModel]) and one that is suitable for image classification (like [BertForSequenceClassification]).