23 lines
574 B
YAML
23 lines
574 B
YAML
dataset_path: "Zombely/wikisource-red"
|
|
pretrained_model_path: "Zombely/pl-donut-v1.1"
|
|
start_model_path: "Zombely/pl-donut-v1.1"
|
|
output_model_path: "Zombely/pl-donut-v1.2"
|
|
wandb_test_name: "pl-donut-v1.2"
|
|
checkpoint_path: "./checkpoint"
|
|
max_length: 768
|
|
image_size: [1260, 960]
|
|
train_config:
|
|
max_epochs: 1
|
|
val_check_interval: 1.0
|
|
check_val_every_n_epoch: 1
|
|
gradient_clip_val: 1.0
|
|
num_training_samples_per_epoch: 800
|
|
lr: 1.0e-4
|
|
train_batch_sizes: [8]
|
|
val_batch_sizes: [1]
|
|
seed: 2023
|
|
num_nodes: 1
|
|
warmup_steps: 10
|
|
result_path: "./result"
|
|
verbose: True
|