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

change the parameter topology of trainer to cost

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