提交 996b1de1 编写于 作者: Y Yi Wang

Rename DataBase into create_data_config_proto

上级 9763761f
...@@ -894,7 +894,7 @@ class MaxOut(Cfg): ...@@ -894,7 +894,7 @@ class MaxOut(Cfg):
self.add_keys(locals()) self.add_keys(locals())
def DataBase(async_load_data=False, def create_data_config_proto(async_load_data=False,
constant_slots=None, constant_slots=None,
data_ratio=1, data_ratio=1,
is_main_data=True, is_main_data=True,
...@@ -924,7 +924,7 @@ def SimpleData(files=None, ...@@ -924,7 +924,7 @@ def SimpleData(files=None,
context_len=None, context_len=None,
buffer_capacity=None, buffer_capacity=None,
**xargs): **xargs):
data_config = DataBase(**xargs) data_config = create_data_config_proto(**xargs)
data_config.type = 'simple' data_config.type = 'simple'
data_config.files = files data_config.files = files
data_config.feat_dim = feat_dim data_config.feat_dim = feat_dim
...@@ -946,7 +946,7 @@ def PyData(files=None, ...@@ -946,7 +946,7 @@ def PyData(files=None,
constant_slots=None, constant_slots=None,
load_thread_num=None, load_thread_num=None,
**xargs): **xargs):
data_config = DataBase(**xargs) data_config = create_data_config_proto(**xargs)
data_config.type = 'py' data_config.type = 'py'
if load_data_module in g_py_module_name_list: if load_data_module in g_py_module_name_list:
...@@ -997,7 +997,7 @@ def ProtoData(files=None, ...@@ -997,7 +997,7 @@ def ProtoData(files=None,
constant_slots=None, constant_slots=None,
load_thread_num=None, load_thread_num=None,
**xargs): **xargs):
data_config = DataBase(**xargs) data_config = create_data_config_proto(**xargs)
if type is None: if type is None:
data_config.type = 'proto' data_config.type = 'proto'
else: else:
...@@ -1036,7 +1036,7 @@ def Data(type, ...@@ -1036,7 +1036,7 @@ def Data(type,
buffer_capacity=None, buffer_capacity=None,
**xargs): **xargs):
data_config = DataBase(**xargs) data_config = create_data_config_proto(**xargs)
data_config.type = type data_config.type = type
data_config.files = files data_config.files = files
data_config.feat_dim = feat_dim data_config.feat_dim = feat_dim
......
...@@ -195,7 +195,7 @@ def define_py_data_sources2(train_list, test_list, module, obj, args=None): ...@@ -195,7 +195,7 @@ def define_py_data_sources2(train_list, test_list, module, obj, args=None):
def py_data2(files, load_data_module, load_data_object, load_data_args, def py_data2(files, load_data_module, load_data_object, load_data_args,
**kwargs): **kwargs):
data = DataBase() data = create_data_config_proto()
data.type = 'py2' data.type = 'py2'
data.files = files data.files = files
data.load_data_module = load_data_module data.load_data_module = load_data_module
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册