from local

This commit is contained in:
dohe0342 2022-12-26 15:15:59 +09:00
parent bc191c29c7
commit 7242f8809d
2 changed files with 1 additions and 0 deletions

View File

@ -1040,6 +1040,7 @@ def train_one_epoch(
scaler.step(optimizer)
scaler.update()
optimizer.zero_grad()
loggin.info('4')
except: # noqa
display_and_save_batch(batch, params=params, sp=sp)