chunked/nltk_chunking/model_doc_layoutlmv2/chunk_21.txt

This means that if you have a batch of
  document images, image should be a tensor of shape (batch_size, 3, 224, 224).