| 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 | |