未验证 提交 02416d0a 编写于 作者: M mapingshuo 提交者: GitHub

Fix 2 bug in train_and_evaluate.py

上级 39cd61e1
...@@ -87,8 +87,8 @@ def evaluate(epoch_id, exe, inference_program, dev_reader, test_reader, fetch_li ...@@ -87,8 +87,8 @@ def evaluate(epoch_id, exe, inference_program, dev_reader, test_reader, fetch_li
def train_and_evaluate(train_reader, def train_and_evaluate(train_reader,
test_reader,
dev_reader, dev_reader,
test_reader,
network, network,
optimizer, optimizer,
global_config, global_config,
...@@ -246,7 +246,10 @@ def main(): ...@@ -246,7 +246,10 @@ def main():
# use cuda or not # use cuda or not
if not global_config.has_member('use_cuda'): if not global_config.has_member('use_cuda'):
global_config.use_cuda = 'CUDA_VISIBLE_DEVICES' in os.environ if 'CUDA_VISIBLE_DEVICES' in os.environ and os.environ['CUDA_VISIBLE_DEVICES'] != '':
global_config.use_cuda = True
else:
global_config.use_cuda = False
global_config.list_config() global_config.list_config()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册