提交 ec23790d 编写于 作者: M mapingshuo

if use_cuda is not set, automatically set it, add a file

上级 11dbb484
...@@ -203,7 +203,6 @@ def main(): ...@@ -203,7 +203,6 @@ def main():
print("net_name: ", args.model_name) print("net_name: ", args.model_name)
net = models.__dict__[args.model_name](global_config) net = models.__dict__[args.model_name](global_config)
global_config.list_config()
# get word_dict # get word_dict
word_dict = utils.getDict(data_type="quora_question_pairs") word_dict = utils.getDict(data_type="quora_question_pairs")
...@@ -232,6 +231,12 @@ def main(): ...@@ -232,6 +231,12 @@ def main():
# define optimizer # define optimizer
optimizer = utils.getOptimizer(global_config) optimizer = utils.getOptimizer(global_config)
# use cuda or not
if not global_config.has_member('use_cuda'):
global_config.use_cuda = 'CUDA_VISIBLE_DEVICES' in os.environ
global_config.list_config()
train_and_evaluate( train_and_evaluate(
train_reader, train_reader,
dev_reader, dev_reader,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册