File size: 358 Bytes
899da1a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
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
|