22 lines
589 B
YAML
22 lines
589 B
YAML
dataset_path: "Zombely/wikisource-small"
|
|
pretrained_model_path: "Zombely/plwiki-fine-tuned-v4"
|
|
start_model_path: "Zombely/plwiki-fine-tuned-v4"
|
|
output_model_path: "Zombely/pl-donut"
|
|
wandb_test_name: "wikisource-small"
|
|
checkpoint_path: "./checkpoint"
|
|
max_length: 768
|
|
image_size: [2560, 1920]
|
|
train_config:
|
|
max_epochs: 1
|
|
val_check_interval: 0.3
|
|
check_val_every_n_epoch: 1
|
|
gradient_clip_val: 1.0
|
|
num_training_samples_per_epoch: 800
|
|
lr: 3.0e-5
|
|
train_batch_sizes: [8]
|
|
val_batch_sizes: [1]
|
|
seed: 2022
|
|
num_nodes: 1
|
|
warmup_steps: 300
|
|
result_path: "./result"
|
|
verbose: True |