From b5c08679a7a5325bdb7591cd63e06e8b64c39939 Mon Sep 17 00:00:00 2001 From: "seujung hwan, Jung" Date: Wed, 16 Jan 2019 07:09:30 +0900 Subject: [PATCH] update Spectral Normalization Code (#13868) * update sn_code * update sn_code --- example/gluon/sn_gan/model.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/example/gluon/sn_gan/model.py b/example/gluon/sn_gan/model.py index 6040adb4eeac..cfd7f93e8dae 100644 --- 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