from local

This commit is contained in:
dohe0342 2022-12-26 15:15:45 +09:00
parent bac2820eb8
commit 7114f16a85
2 changed files with 1 additions and 0 deletions

View File

@ -1020,6 +1020,7 @@ def train_one_epoch(
# NOTE: We use reduction==sum and loss is computed over utterances
# in the batch and there is no normalization to it so far.
loggin.info('1')
scaler.scale(loss).backward()
if params.multi_optim and (batch_idx+1) % params.accum_grads == 0: