Ahmadzei's picture
added 3 more tables for large emb model
5fa1a76
This means that if you have a batch of
document images, image should be a tensor of shape (batch_size, 3, 224, 224).