From c356290c8ddd9037bd08c854b118673983ef6def Mon Sep 17 00:00:00 2001 From: Aditya Soni Date: Sun, 1 Dec 2019 14:08:14 +0530 Subject: [PATCH] typo fix as per Pytorch v1.1+ --- docs/source/migration.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/source/migration.md b/docs/source/migration.md index d04b66d5e4..f50d1dff0a 100644 --- a/docs/source/migration.md +++ b/docs/source/migration.md @@ -104,6 +104,6 @@ for batch in train_data: loss = model(batch) loss.backward() torch.nn.utils.clip_grad_norm_(model.parameters(), max_grad_norm) # Gradient clipping is not in AdamW anymore (so you can use amp without issue) - scheduler.step() optimizer.step() + scheduler.step() ```