提交 796914ef 编写于 作者: X xixiaoyao

refine default settings

上级 b8b478f0
...@@ -28,7 +28,7 @@ class Reader(reader): ...@@ -28,7 +28,7 @@ class Reader(reader):
reader = ClassifyReader(config['vocab_path'], reader = ClassifyReader(config['vocab_path'],
max_seq_len=config['max_seq_len'], max_seq_len=config['max_seq_len'],
do_lower_case=config.get('do_lower_case', False), do_lower_case=config.get('do_lower_case', True),
for_cn=config.get('for_cn', False), for_cn=config.get('for_cn', False),
random_seed=config.get('seed', None)) random_seed=config.get('seed', None))
self._reader = reader self._reader = reader
......
...@@ -40,7 +40,7 @@ class Reader(reader): ...@@ -40,7 +40,7 @@ class Reader(reader):
if phase == 'train': if phase == 'train':
self._input_file = config['train_file'] self._input_file = config['train_file']
self._num_epochs = None # 防止iteartor终止 self._num_epochs = None # 防止iteartor终止
self._shuffle = config.get('shuffle', False) self._shuffle = config.get('shuffle', True)
self._shuffle_buffer = config.get('shuffle_buffer', 5000) self._shuffle_buffer = config.get('shuffle_buffer', 5000)
elif phase == 'eval': elif phase == 'eval':
self._input_file = config['dev_file'] self._input_file = config['dev_file']
......
...@@ -43,7 +43,7 @@ class Reader(reader): ...@@ -43,7 +43,7 @@ class Reader(reader):
self._input_file = config['train_file'] self._input_file = config['train_file']
# self._num_epochs = config['num_epochs'] # self._num_epochs = config['num_epochs']
self._num_epochs = None # 防止iteartor终止 self._num_epochs = None # 防止iteartor终止
self._shuffle = config.get('shuffle', False) self._shuffle = config.get('shuffle', True)
self._shuffle_buffer = config.get('shuffle_buffer', 5000) self._shuffle_buffer = config.get('shuffle_buffer', 5000)
if phase == 'eval': if phase == 'eval':
self._input_file = config['dev_file'] self._input_file = config['dev_file']
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册