ms180 commited on
Commit
6759df9
1 Parent(s): 107649c

Update finetune.py

Browse files
Files changed (1) hide show
  1. finetune.py +4 -0
finetune.py CHANGED
@@ -126,6 +126,10 @@ class CustomFinetuneModel(ESPnetS2TModel):
126
 
127
  def finetune_model(lang, task, tempdir_path, log_every, max_epoch, scheduler, warmup_steps, optimizer, learning_rate, weight_decay):
128
  """Main function for finetuning the model."""
 
 
 
 
129
  print("Start loading dataset...")
130
  if len(tempdir_path) == 0:
131
  raise gr.Error("Please upload a zip file first.")
 
126
 
127
  def finetune_model(lang, task, tempdir_path, log_every, max_epoch, scheduler, warmup_steps, optimizer, learning_rate, weight_decay):
128
  """Main function for finetuning the model."""
129
+
130
+ if not os.path.exists(f"{tempdir_path}/ref.txt"):
131
+ raise gr.Error("Please run Baseline first.")
132
+
133
  print("Start loading dataset...")
134
  if len(tempdir_path) == 0:
135
  raise gr.Error("Please upload a zip file first.")