提交 43f6cae2 编写于 作者: littletomatodonkey's avatar littletomatodonkey

fix cpu train

上级 2d5a8143
...@@ -98,7 +98,8 @@ def check_config(config): ...@@ -98,7 +98,8 @@ def check_config(config):
""" """
check.check_version() check.check_version()
mode = config.get('mode', 'train') use_gpu = config.get('use_gpu', True)
if use_gpu:
check.check_gpu() check.check_gpu()
architecture = config.get('ARCHITECTURE') architecture = config.get('ARCHITECTURE')
...@@ -110,6 +111,7 @@ def check_config(config): ...@@ -110,6 +111,7 @@ def check_config(config):
classes_num = config.get('classes_num') classes_num = config.get('classes_num')
check.check_classes_num(classes_num) check.check_classes_num(classes_num)
mode = config.get('mode', 'train')
if mode.lower() == 'train': if mode.lower() == 'train':
check.check_function_params(config, 'LEARNING_RATE') check.check_function_params(config, 'LEARNING_RATE')
check.check_function_params(config, 'OPTIMIZER') check.check_function_params(config, 'OPTIMIZER')
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册