donut-handwriting / config.yaml
srsani's picture
10a1b3169b6325c9ff48f6409c0d5dc57e85fa882b19fb1c5ebe80c02b8b5194
d4d9786
raw
history blame
No virus
574 Bytes
resume_from_checkpoint_path: None
result_path: './result'
pretrained_model_name_or_path: 'naver-clova-ix/donut-base'
dataset_name_or_paths:
- 'dataset/nist/'
sort_json_key: False
train_batch_sizes:
- 1
val_batch_sizes:
- 1
input_size:
- 1280
- 960
max_length: 768
align_long_axis: False
num_nodes: 1
seed: 2022
lr: 3e-05
warmup_steps: 300
num_training_samples_per_epoch: 800
max_epochs: 30
max_steps: -1
num_workers: 8
val_check_interval: 1.0
check_val_every_n_epoch: 3
gradient_clip_val: 1.0
verbose: True
exp_name: 'train_nist'
exp_version: '20230716_080527'