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

debug

parent c1348ba0
......@@ -1932,7 +1932,6 @@ def new_train_epoch(model,
torch.nn.utils.clip_grad_norm_(model.parameters(), 1.)
scaler.step(optimizer)
scaler.update()
optimizer.step()
else:
loss.backward()
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