Spaces:
Runtime error
Runtime error
Update train_gpt_xtts.py
Browse files- 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__":
|