Spaces:
Runtime error
Runtime error
Update train_gpt_xtts.py
Browse files- train_gpt_xtts.py +1 -0
train_gpt_xtts.py
CHANGED
|
@@ -179,6 +179,7 @@ def main():
|
|
| 179 |
eval_samples=eval_samples,
|
| 180 |
)
|
| 181 |
# trainer.fit()
|
|
|
|
| 182 |
max_epochs = 5
|
| 183 |
for epoch in range(max_epochs):
|
| 184 |
trainer.train_epoch() # entrena una época
|
|
|
|
| 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
|