clip_grad_norm is now deprecated

hellocsi · Nov 9, 2018 · a16173f · a16173f
1 parent c48008b
commit a16173f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tutorials/02-intermediate/language_model/main.py b/tutorials/02-intermediate/language_model/main.py
@@ -3,7 +3,7 @@
 import torch
 import torch.nn as nn
 import numpy as np
-from torch.nn.utils import clip_grad_norm
+from torch.nn.utils import clip_grad_norm_
 from data_utils import Dictionary, Corpus
 
 
@@ -78,7 +78,7 @@ def detach(states):
         # Backward and optimize
         model.zero_grad()
         loss.backward()
-        clip_grad_norm(model.parameters(), 0.5)
+        clip_grad_norm_(model.parameters(), 0.5)
         optimizer.step()
 
         step = (i+1) // seq_length