Update main.py (pytorch#68)

simopal6 · Feb 16, 2017 · ec4802c · ec4802c
1 parent f6770fa
commit ec4802c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/word_language_model/main.py b/word_language_model/main.py
@@ -88,7 +88,7 @@ def clip_gradient(model, clip):
         modulenorm = p.grad.data.norm()
         totalnorm += modulenorm ** 2
     totalnorm = math.sqrt(totalnorm)
-    return min(1, args.clip / (totalnorm + 1e-6))
+    return min(1, clip / (totalnorm + 1e-6))
 
 
 def repackage_hidden(h):