sob111 commited on
Commit
013568d
·
verified ·
1 Parent(s): 7cf25ae

Update train_gpt_xtts.py

Browse files
Files changed (1) hide show
  1. train_gpt_xtts.py +6 -1
train_gpt_xtts.py CHANGED
@@ -179,7 +179,12 @@ def main():
179
  train_samples=train_samples,
180
  eval_samples=eval_samples,
181
  )
182
- trainer.fit()
 
 
 
 
 
183
 
184
 
185
  if __name__ == "__main__":
 
179
  train_samples=train_samples,
180
  eval_samples=eval_samples,
181
  )
182
+ # trainer.fit()
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
 
190
  if __name__ == "__main__":