solve qval

WellyZhang · Nov 22, 2016 · 63af7ce · 63af7ce
1 parent 008b3cd
commit 63af7ce
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/qfuncs.py b/qfuncs.py
@@ -45,7 +45,7 @@ def get_loss_symbols(self):
     def define_loss(self, loss_exp):
 
         self.loss = mx.symbol.MakeLoss(loss_exp, name="qfunc_loss")
-        self.loss = mx.symbol.Group([self.loss, self.qval])
+        self.loss = mx.symbol.Group([self.loss, mx.symbol.BlockGrad(self.qval)])
 
     def define_exe(self, ctx, init, updater, input_shapes=None, args=None, 
                     grad_req=None):