Spaces:
Runtime error
Runtime error
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)
|