Fix style issues.

This commit is contained in:
Fangjun Kuang 2022-05-05 12:53:52 +08:00
parent a0dbfba77d
commit c0ea5808bb

View File

@ -691,7 +691,7 @@ def train_one_epoch(
scaler.step(optimizer)
scaler.update()
optimizer.zero_grad()
except:
except: # noqa
display_and_save_batch(batch, params=params, sp=sp)
raise