Skip to content

Commit 20d9220

Browse files
committed
change the parameter topology of trainer to cost
1 parent 4042797 commit 20d9220

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

demo/mnist/api_train_v2.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@ def event_handler(event):
4141

4242
trainer.train(
4343
train_data_reader=train_reader,
44-
topology=cost,
44+
cost=cost,
4545
parameters=parameters,
4646
event_handler=event_handler,
4747
batch_size=32, # batch size should be refactor in Data reader

python/paddle/v2/trainer.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ def __init__(self, update_equation):
6262

6363
def train(self,
6464
train_data_reader,
65-
topology,
65+
cost,
6666
parameters,
6767
num_passes=1,
6868
test_data_reader=None,
@@ -73,7 +73,7 @@ def train(self,
7373
Training method. Will train num_passes of input data.
7474
7575
:param train_data_reader:
76-
:param topology: cost layers, use one or more Layers to represent it.
76+
:param cost: cost layers, to be optimized.
7777
:param parameters: The parameter pools.
7878
:param num_passes: The total train passes.
7979
:param test_data_reader:
@@ -86,7 +86,7 @@ def train(self,
8686
if event_handler is None:
8787
event_handler = default_event_handler
8888

89-
topology = Topology(topology)
89+
topology = Topology(cost)
9090

9191
__check_train_args__(**locals())
9292

0 commit comments

Comments
 (0)