search

clovaai / donut

Official Implementation of OCR-free Document Understanding Transformer (Donut) and Synthetic Document Generator (SynthDoG), ECCV 2022
https://arxiv.org/abs/2111.15664
MIT License
5.75k stars 466 forks source link

multi gpus, validation error in training stage #137

Open CCchenxiaoxue opened 1 year ago

CCchenxiaoxue commented 1 year ago

I tried finetuning DONUT on RVLCDIP with 8 V100 GPUs. It occurs error when validation in training stage as follows.

RuntimeError: All input tensors must be on the same device. Received cuda:2 and cuda:0

how can i fix it? thanks for your help~