From 43f6cae208296ac05c41f8c5be3f34565b22772c Mon Sep 17 00:00:00 2001
From: littletomatodonkey <dazhiningsibuqu@163.com>
Date: Wed, 21 Oct 2020 14:41:54 +0000
Subject: [PATCH] fix cpu train

---
 ppcls/utils/config.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/ppcls/utils/config.py b/ppcls/utils/config.py
index 25a918afe..3bde9dfbf 100644
--- a/ppcls/utils/config.py
+++ b/ppcls/utils/config.py
@@ -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')