Commit 0054678b authored by Anthony Larcher's avatar Anthony Larcher
Browse files


parent c1348ba0
...@@ -1932,7 +1932,6 @@ def new_train_epoch(model, ...@@ -1932,7 +1932,6 @@ def new_train_epoch(model,
torch.nn.utils.clip_grad_norm_(model.parameters(), 1.) torch.nn.utils.clip_grad_norm_(model.parameters(), 1.)
scaler.step(optimizer) scaler.step(optimizer)
scaler.update() scaler.update()
else: else:
loss.backward() loss.backward()
optimizer.step() optimizer.step()
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment