mrfakename commited on
Commit
48c3700
1 Parent(s): a674527

Sync from GitHub repo

Browse files

This Space is synced from the GitHub repo: https://github.com/SWivid/F5-TTS. Please submit contributions to the Space there

Files changed (1) hide show
  1. finetune_gradio.py +5 -1
finetune_gradio.py CHANGED
@@ -294,7 +294,9 @@ def start_training(dataset_name="",
294
  f"--last_per_steps {last_per_steps} " \
295
  f"--dataset_name {dataset_name}"
296
  if finetune:cmd += f" --finetune {finetune}"
297
- print(cmd)
 
 
298
  try:
299
  # Start the training process
300
  training_process = subprocess.Popen(cmd, shell=True)
@@ -525,6 +527,8 @@ def calculate_train(name_project,batch_size_type,max_samples,learning_rate,num_w
525
 
526
  if samples<64:
527
  max_samples = int(samples * 0.25)
 
 
528
 
529
  num_warmup_updates = int(samples * 0.10)
530
  save_per_updates = int(samples * 0.25)
 
294
  f"--last_per_steps {last_per_steps} " \
295
  f"--dataset_name {dataset_name}"
296
  if finetune:cmd += f" --finetune {finetune}"
297
+
298
+ print(cmd)
299
+
300
  try:
301
  # Start the training process
302
  training_process = subprocess.Popen(cmd, shell=True)
 
527
 
528
  if samples<64:
529
  max_samples = int(samples * 0.25)
530
+ else:
531
+ max_samples = 64
532
 
533
  num_warmup_updates = int(samples * 0.10)
534
  save_per_updates = int(samples * 0.25)