diff --git a/config-train.yaml b/config-train.yaml index 67f01c4..19917ef 100644 --- a/config-train.yaml +++ b/config-train.yaml @@ -1,6 +1,6 @@ dataset_path: "Zombely/wikisource-small" -pretrained_model_path: "Zombely/plwiki-proto-fine-tuned-v3.2" -start_model_path: "Zombely/plwiki-proto-fine-tuned-v3.2" +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"