from local

This commit is contained in:
dohe0342 2022-12-10 14:04:01 +09:00
parent 8deeece377
commit cb0aa1247d
2 changed files with 1 additions and 1 deletions

View File

@ -1089,7 +1089,7 @@ def train_one_epoch(
wb.log({"valid/pruned_loss": valid_info["pruned_loss"]*numel})
wb.log({"valid/loss": valid_info["loss"]*numel})
loss_value = tot_loss["loss"] / tot_loss["frames"]
loss_value = tot_loss["loss"] / tot_loss["utterances"]
params.train_loss = loss_value
if params.train_loss < params.best_train_loss:
params.best_train_epoch = params.cur_epoch