提交 20d9220c 编写于 作者: Q qiaolongfei

change the parameter topology of trainer to cost

上级 40427979
...@@ -41,7 +41,7 @@ def main(): ...@@ -41,7 +41,7 @@ def main():
trainer.train( trainer.train(
train_data_reader=train_reader, train_data_reader=train_reader,
topology=cost, cost=cost,
parameters=parameters, parameters=parameters,
event_handler=event_handler, event_handler=event_handler,
batch_size=32, # batch size should be refactor in Data reader batch_size=32, # batch size should be refactor in Data reader
......
...@@ -62,7 +62,7 @@ class SGD(ITrainer): ...@@ -62,7 +62,7 @@ class SGD(ITrainer):
def train(self, def train(self,
train_data_reader, train_data_reader,
topology, cost,
parameters, parameters,
num_passes=1, num_passes=1,
test_data_reader=None, test_data_reader=None,
...@@ -73,7 +73,7 @@ class SGD(ITrainer): ...@@ -73,7 +73,7 @@ class SGD(ITrainer):
Training method. Will train num_passes of input data. Training method. Will train num_passes of input data.
:param train_data_reader: :param train_data_reader:
:param topology: cost layers, use one or more Layers to represent it. :param cost: cost layers, to be optimized.
:param parameters: The parameter pools. :param parameters: The parameter pools.
:param num_passes: The total train passes. :param num_passes: The total train passes.
:param test_data_reader: :param test_data_reader:
...@@ -86,7 +86,7 @@ class SGD(ITrainer): ...@@ -86,7 +86,7 @@ class SGD(ITrainer):
if event_handler is None: if event_handler is None:
event_handler = default_event_handler event_handler = default_event_handler
topology = Topology(topology) topology = Topology(cost)
__check_train_args__(**locals()) __check_train_args__(**locals())
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册