File tree Expand file tree Collapse file tree 2 files changed +4
-4
lines changed Expand file tree Collapse file tree 2 files changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -41,7 +41,7 @@ def event_handler(event):
41
41
42
42
trainer .train (
43
43
train_data_reader = train_reader ,
44
- topology = cost ,
44
+ cost = cost ,
45
45
parameters = parameters ,
46
46
event_handler = event_handler ,
47
47
batch_size = 32 , # batch size should be refactor in Data reader
Original file line number Diff line number Diff line change @@ -62,7 +62,7 @@ def __init__(self, update_equation):
62
62
63
63
def train (self ,
64
64
train_data_reader ,
65
- topology ,
65
+ cost ,
66
66
parameters ,
67
67
num_passes = 1 ,
68
68
test_data_reader = None ,
@@ -73,7 +73,7 @@ def train(self,
73
73
Training method. Will train num_passes of input data.
74
74
75
75
: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 .
77
77
:param parameters: The parameter pools.
78
78
:param num_passes: The total train passes.
79
79
:param test_data_reader:
@@ -86,7 +86,7 @@ def train(self,
86
86
if event_handler is None :
87
87
event_handler = default_event_handler
88
88
89
- topology = Topology (topology )
89
+ topology = Topology (cost )
90
90
91
91
__check_train_args__ (** locals ())
92
92
You can’t perform that action at this time.
0 commit comments