File tree Expand file tree Collapse file tree 3 files changed +15
-2
lines changed
python/paddle/trainer_config_helpers Expand file tree Collapse file tree 3 files changed +15
-2
lines changed Original file line number Diff line number Diff line change @@ -139,7 +139,7 @@ def __is_splitable__(o):
139
139
test_obj = obj
140
140
train_obj = obj
141
141
if __is_splitable__ (obj ):
142
- train_module , test_module = module
142
+ train_obj , test_obj = obj
143
143
144
144
if args is None :
145
145
args = ""
Original file line number Diff line number Diff line change @@ -11,7 +11,8 @@ test_sequence_pooling test_lstmemory_layer test_grumemory_layer
11
11
last_first_seq test_expand_layer test_ntm_layers test_hsigmoid
12
12
img_layers img_trans_layers util_layers simple_rnn_layers unused_layers test_cost_layers
13
13
test_rnn_group shared_fc shared_lstm test_cost_layers_with_weight
14
- test_bilinear_interp test_maxout test_bi_grumemory math_ops)
14
+ test_bilinear_interp test_maxout test_bi_grumemory math_ops
15
+ test_spilit_datasource)
15
16
16
17
17
18
for conf in ${configs[*]}
Original file line number Diff line number Diff line change
1
+ from paddle .trainer_config_helpers import *
2
+
3
+ define_py_data_sources2 (train_list = "train.list" ,
4
+ test_list = "test.list" ,
5
+ module = ["a" , "b" ],
6
+ obj = ("c" , "d" ))
7
+ settings (
8
+ learning_rate = 1e-3 ,
9
+ batch_size = 1000
10
+ )
11
+
12
+ outputs (data_layer (name = "a" , size = 10 ))
You can’t perform that action at this time.
0 commit comments