from local
This commit is contained in:
parent
bc191c29c7
commit
7242f8809d
Binary file not shown.
@ -1040,6 +1040,7 @@ def train_one_epoch(
|
|||||||
scaler.step(optimizer)
|
scaler.step(optimizer)
|
||||||
scaler.update()
|
scaler.update()
|
||||||
optimizer.zero_grad()
|
optimizer.zero_grad()
|
||||||
|
loggin.info('4')
|
||||||
|
|
||||||
except: # noqa
|
except: # noqa
|
||||||
display_and_save_batch(batch, params=params, sp=sp)
|
display_and_save_batch(batch, params=params, sp=sp)
|
||||||
|
|||||||
Reference in New Issue
Block a user