提交 33b81648 编写于 作者: Y Yu Yang

Fix bug in multple objects in define_py_sources

上级 8295eb91
...@@ -139,7 +139,7 @@ def define_py_data_sources(train_list, test_list, module, obj, args=None, ...@@ -139,7 +139,7 @@ def define_py_data_sources(train_list, test_list, module, obj, args=None,
test_obj = obj test_obj = obj
train_obj = obj train_obj = obj
if __is_splitable__(obj): if __is_splitable__(obj):
train_module, test_module = module train_obj, test_obj = obj
if args is None: if args is None:
args = "" args = ""
......
...@@ -11,7 +11,8 @@ test_sequence_pooling test_lstmemory_layer test_grumemory_layer ...@@ -11,7 +11,8 @@ test_sequence_pooling test_lstmemory_layer test_grumemory_layer
last_first_seq test_expand_layer test_ntm_layers test_hsigmoid last_first_seq test_expand_layer test_ntm_layers test_hsigmoid
img_layers img_trans_layers util_layers simple_rnn_layers unused_layers test_cost_layers img_layers img_trans_layers util_layers simple_rnn_layers unused_layers test_cost_layers
test_rnn_group shared_fc shared_lstm test_cost_layers_with_weight test_rnn_group shared_fc shared_lstm test_cost_layers_with_weight
test_bilinear_interp test_maxout test_bi_grumemory math_ops) test_bilinear_interp test_maxout test_bi_grumemory math_ops
test_spilit_datasource)
for conf in ${configs[*]} for conf in ${configs[*]}
......
from paddle.trainer_config_helpers import *
define_py_data_sources2(train_list="train.list",
test_list="test.list",
module=["a", "b"],
obj=("c", "d"))
settings(
learning_rate=1e-3,
batch_size=1000
)
outputs(data_layer(name="a", size=10))
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册