uartimcs commited on
Commit
b402d3b
·
verified ·
1 Parent(s): 71e61c1

Delete config/train_rvlcdip.yaml

Browse files
Files changed (1) hide show
  1. config/train_rvlcdip.yaml +0 -23
config/train_rvlcdip.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/rvlcdip"] # should be prepared from https://www.cs.cmu.edu/~aharley/rvl-cdip/
5
- sort_json_key: True
6
- train_batch_sizes: [2]
7
- val_batch_sizes: [4]
8
- input_size: [2560, 1920]
9
- max_length: 8
10
- align_long_axis: False
11
- # num_nodes: 8 # memo: donut-base-finetuned-rvlcdip was trained with 8 nodes
12
- num_nodes: 1
13
- seed: 2022
14
- lr: 2e-5
15
- warmup_steps: 10000
16
- num_training_samples_per_epoch: 320000
17
- max_epochs: 100
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: 1.0
23
- verbose: True