From 5f5e5c32e50099df02e516647192c5a07b588f99 Mon Sep 17 00:00:00 2001 From: qiaolongfei Date: Tue, 28 Feb 2017 10:33:46 +0800 Subject: [PATCH] modify train in ITrainer --- python/paddle/v2/trainer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/paddle/v2/trainer.py b/python/paddle/v2/trainer.py index be33b910807..2aeddaff897 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: -- GitLab