s444415/donut
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

config

Browse Source
This commit is contained in:
Michał Kozłowski 2023-01-24 18:14:10 +01:00
parent 3ad876ea69
commit 9ba7e24ebd
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
config-train.yaml
View File

@ -1,14 +1,14 @@
dataset_path: "Zombely/fiszki-ocr-train" dataset_path: "Zombely/wikisource-small"
pretrained_model_path: "Zombely/plwiki-proto-fine-tuned-v2" pretrained_model_path: "Zombely/plwiki-proto-fine-tuned-v3.2"
start_model_path: "Zombely/plwiki-proto-fine-tuned-v2" start_model_path: "Zombely/plwiki-proto-fine-tuned-v3.2"
output_model_path: "Zombely/plwiki-proto-fine-tuned-v3" output_model_path: "Zombely/pl-donut"
wandb_test_name: "fiszki-ocr-fine-tune" wandb_test_name: "wikisource-small"
checkpoint_path: "./checkpoint" checkpoint_path: "./checkpoint"
max_length: 768 max_length: 768
image_size: [2560, 1920] image_size: [2560, 1920]
train_config: train_config:
max_epochs: 1 max_epochs: 1
val_check_interval: 0.5 val_check_interval: 0.3
check_val_every_n_epoch: 1 check_val_every_n_epoch: 1
gradient_clip_val: 1.0 gradient_clip_val: 1.0
num_training_samples_per_epoch: 800 num_training_samples_per_epoch: 800