我正在尝试使用自定义数据集运行layout-xlm,遇到了这个问题。
LayoutLMv2的后端问题。我尝试了以下方法，但没有解决问题。
1)我尝试通过修改data_collactor来解决这个问题。

batch['image'] = ImageList.from_tensors([feature['image'] for feature in features], 32)

2)在modelling_layoutlmv2中：

def forward(self, images):
    images_input = (
        (images if torch.is_tensor(images) else images.tensor) - self.pixel_mean
    ) / self.pixel_std