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

fixup! Fix aux unit tests

This commit is contained in:
Enno Hermann 2023-12-08 08:37:28 +01:00
parent 32e777ae7a
commit e0ef13cd70
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
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):