提交 ec23790d 编写于 作者: M mapingshuo

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

上级 11dbb484
......@@ -203,7 +203,6 @@ def main():
print("net_name: ", args.model_name)
net = models.__dict__[args.model_name](global_config)
global_config.list_config()
# get word_dict
word_dict = utils.getDict(data_type="quora_question_pairs")
......@@ -231,6 +230,12 @@ def main():
# define optimizer
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_reader,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册