Merge pull request #74 from WuHaobo/googlenet

fix mixup while using GoogleNet

ppcls/utils/check.py

@@ -16,6 +16,7 @@ from __future__ import absolute_import
 from __future__ import division
 from __future__ import print_function
 
+import os
 import sys
 
 import paddle.fluid as fluid
@@ -36,7 +37,7 @@ def check_version():
 
     try:
         fluid.require_version('1.7.0')
-    except Exception as e:
+    except Exception:
         logger.error(err)
         sys.exit(1)
 
@@ -62,7 +63,8 @@ def check_architecture(architecture):
     assert isinstance(architecture, dict), \
         ("the type of architecture({}) should be dict". format(architecture))
     assert "name" in architecture, \
-            ("name must be in the architecture keys, just contains: {}". format(architecture.keys()))
+        ("name must be in the architecture keys, just contains: {}". format(
+            architecture.keys()))
 
     similar_names = similar_architectures(architecture["name"],
                                           get_architectures())
@@ -83,7 +85,8 @@ def check_mix(architecture, use_mix=False):
     err = "Cannot use mix processing in GoogLeNet, " \
           "please set use_mix = False."
     try:
-        if architecture["name"] == "GoogLeNet": assert use_mix == False
+        if architecture["name"] == "GoogLeNet":
+            assert use_mix is not True
     except AssertionError:
         logger.error(err)
         sys.exit(1)

ppcls/utils/config.py

@@ -100,7 +100,7 @@ def check_config(config):
     architecture = config.get('ARCHITECTURE')
     check.check_architecture(architecture)
 
-    use_mix = config.get('use_mix')
+    use_mix = config.get('use_mix', False)
     check.check_mix(architecture, use_mix)
 
     classes_num = config.get('classes_num')