modify train in ITrainer

8 years ago · 5f5e5c32e5
parent 20d9220c91
commit 5f5e5c32e5
1 changed files with 2 additions and 2 deletions
--- a/python/paddle/v2/trainer.py
+++ b/python/paddle/v2/trainer.py
@ -29,7 +29,7 @@ class ITrainer(object):

    def train(self,
              train_data_reader,
-              topology,
+              cost,
              parameters,
              test_data_reader=None,
              event_handler=None):
@ -37,7 +37,7 @@ class ITrainer(object):
        train method.

        :param train_data_reader:
-        :param topology:
+        :param cost:
        :param parameters:
        :param test_data_reader:
        :param event_handler: