batch_size: 16
data_parallel: false
dataset_name: MP-DocVQA
device: cuda
eval_start: false
images_dir: /data/users/rperez/MP-DocVQA/images/
imdb_dir: /data/users/rperez/imdb/collection
lr: 2e-4
max_source_length: 512
model_name: VT5
model_weights: rubentito/t5-base-mpdocvqa
page_retrieval: Oracle
save_dir: save/
seed: 42
train_epochs: 10
visual_module:
  model: dit
  model_weights: microsoft/dit-base-finetuned-rvlcdip
warmup_iterations: 1000