Skip to content

Commit 3fd13e7

Browse files
authored
Merge pull request #1556 from helinwang/sum
change sumCost to sum
2 parents 0bf542e + dbffda2 commit 3fd13e7

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

python/paddle/v2/trainer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -101,7 +101,7 @@ def train(self, reader, num_passes=1, event_handler=None, reader_dict=None):
101101
for each_param in self.__gradient_machine__.getNonStaticParameters(
102102
):
103103
updater.update(each_param)
104-
cost_sum = out_args.sumCosts()
104+
cost_sum = out_args.sum()
105105
cost = cost_sum / len(data_batch)
106106
updater.finishBatch(cost)
107107
batch_evaluator.finish()
@@ -137,7 +137,7 @@ def test(self, reader, reader_dict=None):
137137
num_samples += len(data_batch)
138138
self.__gradient_machine__.forward(
139139
feeder(data_batch), out_args, api.PASS_TEST)
140-
total_cost += out_args.sumCosts()
140+
total_cost += out_args.sum()
141141
self.__gradient_machine__.eval(evaluator)
142142

143143
evaluator.finish()

0 commit comments

Comments
 (0)