提交 69bf77fd 编写于 作者: L liaogang

fix trainer v2 getNonStaticParameters

上级 ce3a399d
......@@ -113,7 +113,7 @@ class SGD(ITrainer):
gm.forwardBackward(feeder(data_batch), out_args, pass_type)
gm.eval(pass_evaluator)
gm.eval(batch_evaluator)
for each_param in gm.getParameters():
for each_param in gm.getNonStaticParameters():
updater.update(each_param)
# Get cost. We use numpy to calculate total cost for this batch.
cost_vec = out_args.getSlotValue(0)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册