| base_dir=/path/to/dimt_ocr_free | |
| dataset_dir=/path/to/DoTA_dataset | |
| donut_dir=/path/to/donut-base | |
| nougat_dir=/path/to/nougat-small | |
| export CUDA_VISIBLE_DEVICES=0,1 | |
| accelerate launch \ | |
| --main_process_port 12345 \ | |
| --num_processes 2 \ | |
| --num_machines 1 train.py \ | |
| --base_dir $base_dir \ | |
| --dataset_dir $dataset_dir \ | |
| --donut_dir $donut_dir \ | |
| --nougat_dir $nougat_dir |