sob111 commited on
Commit
0e4ba1e
·
verified ·
1 Parent(s): 0c46ac0

Update train_gpt_xtts.py

Browse files
Files changed (1) hide show
  1. train_gpt_xtts.py +7 -6
train_gpt_xtts.py CHANGED
@@ -178,12 +178,13 @@ def main():
178
  train_samples=train_samples,
179
  eval_samples=eval_samples,
180
  )
181
- # trainer.fit()
182
- trainer.init_training()
183
- max_epochs = 5
184
- for epoch in range(max_epochs):
185
- trainer.train_epoch() # entrena una época
186
- trainer.eval_epoch() # evalúa
 
187
 
188
 
189
 
 
178
  train_samples=train_samples,
179
  eval_samples=eval_samples,
180
  )
181
+
182
+ trainer.fit(max_steps=10)
183
+ #trainer.init_training()
184
+ #max_epochs = 5
185
+ #for epoch in range(max_epochs):
186
+ # trainer.train_epoch() # entrena una época
187
+ # trainer.eval_epoch() # evalúa
188
 
189
 
190