mirrors/PaddleClas
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleClas.git synced 2025-06-03 21:55:06 +08:00
Code Issues Projects Releases Wiki Activity

fix cpu train

Browse Source
This commit is contained in:
littletomatodonkey 2020-10-21 14:41:54 +00:00
parent 2d5a81437f
commit 43f6cae208
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
ppcls/utils/config.py
View File

@ -98,8 +98,9 @@ def check_config(config):
"""
check.check_version()
mode = config.get('mode', 'train')
check.check_gpu()
use_gpu = config.get('use_gpu', True)
if use_gpu:
check.check_gpu()
architecture = config.get('ARCHITECTURE')
check.check_architecture(architecture)
@ -110,6 +111,7 @@ def check_config(config):
classes_num = config.get('classes_num')
check.check_classes_num(classes_num)
mode = config.get('mode', 'train')
if mode.lower() == 'train':
check.check_function_params(config, 'LEARNING_RATE')
check.check_function_params(config, 'OPTIMIZER')