ibcplateformes Claude Opus 4.6 commited on
Commit
81bdd2b
·
1 Parent(s): 447db6e

Restore full quality params now that GPU is available

Browse files

- Max slices: 80 (use all audio)
- Batch size: 8
- Default epochs: 20
- Slider max: 50

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>

Files changed (2) hide show
  1. app.py +4 -4
  2. pipeline/training.py +3 -3
app.py CHANGED
@@ -112,7 +112,7 @@ def train_voice_model(audio_file, model_name, epochs, progress=gr.Progress()):
112
  model_name=model_name,
113
  epochs=int(epochs),
114
  sample_rate=40000,
115
- batch_size=4,
116
  progress_callback=progress_callback,
117
  )
118
 
@@ -295,10 +295,10 @@ with gr.Blocks(
295
  )
296
  train_epochs = gr.Slider(
297
  minimum=5,
298
- maximum=30,
299
- value=10,
300
  step=5,
301
- label="Nombre d'époques (utilisé uniquement avec GPU)",
302
  )
303
  train_btn = gr.Button(
304
  "Lancer l'entraînement",
 
112
  model_name=model_name,
113
  epochs=int(epochs),
114
  sample_rate=40000,
115
+ batch_size=8,
116
  progress_callback=progress_callback,
117
  )
118
 
 
295
  )
296
  train_epochs = gr.Slider(
297
  minimum=5,
298
+ maximum=50,
299
+ value=20,
300
  step=5,
301
+ label="Nombre d'époques (plus = meilleure qualité, ~3-5 min avec GPU)",
302
  )
303
  train_btn = gr.Button(
304
  "Lancer l'entraînement",
pipeline/training.py CHANGED
@@ -95,7 +95,7 @@ def preprocess(model_name: str, audio_path: str, sample_rate: int = 40000):
95
  segment_len_16k = int(3.5 * 16000)
96
  hop_16k = int(3.0 * 16000)
97
 
98
- MAX_SLICES = 30 # Limit for CPU-feasible training
99
 
100
  n_slices = 0
101
  idx = 0
@@ -229,8 +229,8 @@ def extract_features(model_name: str, sample_rate: int = 40000, f0_method: str =
229
  def train_model(
230
  model_name: str,
231
  sample_rate: int = 40000,
232
- total_epochs: int = 10,
233
- batch_size: int = 4,
234
  ):
235
  """
236
  Train RVC v2 model. Runs IN-PROCESS with mp.Process patched to avoid
 
95
  segment_len_16k = int(3.5 * 16000)
96
  hop_16k = int(3.0 * 16000)
97
 
98
+ MAX_SLICES = 80 # Use all available audio for better quality
99
 
100
  n_slices = 0
101
  idx = 0
 
229
  def train_model(
230
  model_name: str,
231
  sample_rate: int = 40000,
232
+ total_epochs: int = 20,
233
+ batch_size: int = 8,
234
  ):
235
  """
236
  Train RVC v2 model. Runs IN-PROCESS with mp.Process patched to avoid