Spaces:
Sleeping
Sleeping
Update finetune.py
Browse files- finetune.py +4 -2
finetune.py
CHANGED
@@ -128,10 +128,12 @@ class CustomFinetuneModel(ESPnetS2TModel):
|
|
128 |
def finetune_model(lang, task, tempdir_path, log_every, max_epoch, scheduler, warmup_steps, optimizer, learning_rate, weight_decay):
|
129 |
"""Main function for finetuning the model."""
|
130 |
|
131 |
-
|
132 |
-
|
|
|
133 |
|
134 |
print("Start loading dataset...")
|
|
|
135 |
if len(tempdir_path) == 0:
|
136 |
raise gr.Error("Please upload a zip file first.")
|
137 |
|
|
|
128 |
def finetune_model(lang, task, tempdir_path, log_every, max_epoch, scheduler, warmup_steps, optimizer, learning_rate, weight_decay):
|
129 |
"""Main function for finetuning the model."""
|
130 |
|
131 |
+
print("Start generating baseline...")
|
132 |
+
gr.Info("Start generating baseline...")
|
133 |
+
baseline_model(lang, task, tempdir_path)
|
134 |
|
135 |
print("Start loading dataset...")
|
136 |
+
gr.Info("Start Fine-tuning process...")
|
137 |
if len(tempdir_path) == 0:
|
138 |
raise gr.Error("Please upload a zip file first.")
|
139 |
|