Spaces:
Running
Running
Delete config/train_docvqa.yaml
Browse files- config/train_docvqa.yaml +0 -23
config/train_docvqa.yaml
DELETED
@@ -1,23 +0,0 @@
|
|
1 |
-
resume_from_checkpoint_path: null
|
2 |
-
result_path: "./result"
|
3 |
-
pretrained_model_name_or_path: "naver-clova-ix/donut-base"
|
4 |
-
dataset_name_or_paths: ["./dataset/docvqa"] # should be prepared from https://rrc.cvc.uab.es/?ch=17
|
5 |
-
sort_json_key: True
|
6 |
-
train_batch_sizes: [2]
|
7 |
-
val_batch_sizes: [4]
|
8 |
-
input_size: [2560, 1920]
|
9 |
-
max_length: 128
|
10 |
-
align_long_axis: False
|
11 |
-
# num_nodes: 8 # memo: donut-base-finetuned-docvqa was trained with 8 nodes
|
12 |
-
num_nodes: 1
|
13 |
-
seed: 2022
|
14 |
-
lr: 3e-5
|
15 |
-
warmup_steps: 10000
|
16 |
-
num_training_samples_per_epoch: 39463
|
17 |
-
max_epochs: 300
|
18 |
-
max_steps: -1
|
19 |
-
num_workers: 8
|
20 |
-
val_check_interval: 1.0
|
21 |
-
check_val_every_n_epoch: 1
|
22 |
-
gradient_clip_val: 0.25
|
23 |
-
verbose: True
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|