longt5-tglobal-base-mpdocvqa / experiment_config.yml
rubentito's picture
Upload 9 files
899da1a
raw
history blame
358 Bytes
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