batch_size: 8 | |
data_parallel: false | |
dataset_name: MP-DocVQA | |
device: cuda | |
eval_start: true | |
images_dir: /data/users/rperez/MP-DocVQA/images/ | |
imdb_dir: /data/users/rperez/imdb/collection | |
lr: 1e-4 | |
model_name: LongT5 | |
model_weights: tryolabs/long-t5-tglobal-base-blogpost-cqa | |
page_retrieval: Oracle | |
save_dir: save/ | |
seed: 42 | |
train_epochs: 10 | |
warmup_iterations: 1000 | |