提交 44e7de9c 编写于 作者: H helinwang 提交者: GitHub

Merge pull request #1526 from reyoung/feature/fix_merge_error_in_trainer

Fix duplicated forward/backward in trainer.
...@@ -108,9 +108,6 @@ class SGD(ITrainer): ...@@ -108,9 +108,6 @@ class SGD(ITrainer):
pass_evaluator.start() pass_evaluator.start()
updater.startPass() updater.startPass()
for batch_id, data_batch in enumerate(reader()): for batch_id, data_batch in enumerate(reader()):
pass_type = updater.startBatch(len(data_batch))
self.__gradient_machine__.forwardBackward(
feeder(data_batch), out_args, pass_type)
batch_evaluator.start() batch_evaluator.start()
event_handler( event_handler(
v2_event.BeginIteration( v2_event.BeginIteration(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册