Merge pull request yunjey#143 from AutuanLiu/patch01

yunjey · web-flow · commit 4896cefea1eb · 2018-11-09T17:53:38.000+09:00
clip_grad_norm is now deprecated
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
