mirrors
/
PaddleClas
mirror of https://github.com/PaddlePaddle/PaddleClas.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Merge pull request #324 from littletomatodonkey/dyg/fix_cpu_train 

fix cpu train
pull/327/head
littletomatodonkey 2020-10-21 22:46:23 +08:00 committed by GitHub
parent 2d5a81437f 43f6cae208
commit 400c5adf98
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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')