fix sum

WellyZhang · Nov 14, 2016 · ddc09a7 · ddc09a7
1 parent bb007e8
commit ddc09a7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ddpg.py b/ddpg.py
@@ -74,7 +74,7 @@ def init_net(self):
         qval_sym = loss_symbols["qval"]
         yval_sym = loss_symbols["yval"]
         weight_sym_dict = loss_symbols["weight"]
-        reg = self.qfunc_weight_decay * 0.5 * mx.symbol.sum(
+        reg = self.qfunc_weight_decay * 0.5 * sum(
             [mx.symbol.sum(mx.symbol.square(arr))
              for name, arr in weight_sym_dict.items()])
         loss = 1.0 / self.batch_size * mx.symbol.sum(
@@ -108,7 +108,7 @@ def init_net(self):
         loss_symbols = self.policy.get_loss_symbols()
         act_sym = loss_symbols["act"]
         weight_sym_dict = loss["weight"]
-        reg = self.policy_weight_decay * 0.5 * mx.symbol.sum(
+        reg = self.policy_weight_decay * 0.5 * sum(
             [mx.symbol.sum(mx.symbol.square(arr)) 
              for name, arr in weight_sym_dict.items()])
         policy_qval = qval_sym