Spaces:
Running
Running
mrfakename
commited on
Commit
•
48c3700
1
Parent(s):
a674527
Sync from GitHub repo
Browse filesThis Space is synced from the GitHub repo: https://github.com/SWivid/F5-TTS. Please submit contributions to the Space there
- 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 |
-
|
|
|
|
|
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)
|