Using Trainer with LayoutXLM for classification
When using LayoutXLM for classification with the Huggingface Trainer I need to set dataloader_pin_memory as False following the traceback from this error:RuntimeError: cannot pin ‘torch.cuda.LongTensor’ only dense CPU tensors can be pinnedWhen doing that I get the following error… :RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:1 and cuda:0!Any ideas?
2023-05-03T15:24:34Z
Have you solved it?