Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #3423 from idiap/fix-aux-tests 

Fix CI (save best model after 0 steps in tests)
This commit is contained in:
Eren Gölge 2023-12-14 18:00:30 +01:00 committed by GitHub
parent 186cafb34c 99fee6f5ad
commit 55c7063724
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
TTS/bin/train_encoder.py
View File

@ -125,7 +125,7 @@ def evaluation(model, criterion, data_loader, global_step):
def train(model, optimizer, scheduler, criterion, data_loader, eval_data_loader, global_step): def train(model, optimizer, scheduler, criterion, data_loader, eval_data_loader, global_step):
model.train() model.train()
best_loss = float("inf") best_loss = {"train_loss": None, "eval_loss": float("inf")}
avg_loader_time = 0 avg_loader_time = 0
end_time = time.time() end_time = time.time()
for epoch in range(c.epochs): for epoch in range(c.epochs):
@ -248,7 +248,7 @@ def train(model, optimizer, scheduler, criterion, data_loader, eval_data_loader,
) )
# save the best checkpoint # save the best checkpoint
best_loss = save_best_model( best_loss = save_best_model(
eval_loss, {"train_loss": None, "eval_loss": eval_loss},
best_loss, best_loss,
c, c,
model, model,

2
requirements.txt
View File

@ -27,7 +27,7 @@ pandas>=1.4,<2.0
# deps for training # deps for training
matplotlib>=3.7.0 matplotlib>=3.7.0
# coqui stack # coqui stack
trainer>=0.0.32 trainer>=0.0.36
# config management # config management
coqpit>=0.0.16 coqpit>=0.0.16
# chinese g2p deps # chinese g2p deps