#!/bin/bash

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