update Spectral Normalization Code (apache#13868)

* update sn_code * update sn_code
haohuanw · Jun 23, 2019 · b5c0867 · b5c0867
1 parent 374f3c5
commit b5c0867
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/example/gluon/sn_gan/model.py b/example/gluon/sn_gan/model.py
@@ -21,7 +21,7 @@
 
 import mxnet as mx
 from mxnet import nd
-from mxnet import gluon
+from mxnet import gluon, autograd
 from mxnet.gluon import Block
 
 
@@ -68,7 +68,8 @@ def _spectral_norm(self):
         if sigma == 0.:
             sigma = EPSILON
 
-        self.params.setattr('u', _u)
+        with autograd.pause():
+            self.u.set_data(_u)
 
         return w / sigma