Merge pull request #1556 from helinwang/sum

change sumCost to sum
8 years ago · 3fd13e7054
parent 0bf542ec85 dbffda2ff5
commit 3fd13e7054
1 changed files with 2 additions and 2 deletions
--- a/python/paddle/v2/trainer.py
+++ b/python/paddle/v2/trainer.py
@ -101,7 +101,7 @@ class SGD():
                for each_param in self.__gradient_machine__.getNonStaticParameters(
                ):
                    updater.update(each_param)
-                cost_sum = out_args.sumCosts()
+                cost_sum = out_args.sum()
                cost = cost_sum / len(data_batch)
                updater.finishBatch(cost)
                batch_evaluator.finish()
@ -137,7 +137,7 @@ class SGD():
            num_samples += len(data_batch)
            self.__gradient_machine__.forward(
                feeder(data_batch), out_args, api.PASS_TEST)
-            total_cost += out_args.sumCosts()
+            total_cost += out_args.sum()
            self.__gradient_machine__.eval(evaluator)

        evaluator.finish()