from local

This commit is contained in:
dohe0342 2023-01-09 20:27:02 +09:00
parent 61a7fdcdb4
commit b6888bd0de
2 changed files with 1 additions and 1 deletions

View File

@ -682,7 +682,7 @@ def compute_loss(
''' '''
mse_loss = model( mse_loss = model(
x=feature, x=feature,
x_lens=feautre_lens, x_lens=feature_lens,
) )
# after the main warmup step, we keep pruned_loss_scale small # after the main warmup step, we keep pruned_loss_scale small