diff --git a/python/paddle/v2/trainer.py b/python/paddle/v2/trainer.py
index be33b910807ff30eb8ea9b37d5b172a556a1c3f5..2aeddaff89748ef6c769e1793345f5b143b941c6 100644
--- 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: