[ppfleetx] Remove reference of third lib 'six'

codestyle/docstring_checker.py

@@ -13,7 +13,6 @@
 # limitations under the License.
 """DocstringChecker is used to check python doc string's style."""
 
-import six
 import astroid
 
 from pylint.checkers import BaseChecker, utils

ppfleetx/core/engine/eager_engine.py

@@ -305,6 +305,9 @@ def _train_one_epoch(self,
                 train_step_start = get_timestamp()
                 train_losses = []
 
+            if self._lr_scheduler is not None and self._lr_scheduler_mode == 'step':
+                self._lr_scheduler.step()
+
             self._optimizer.clear_grad()
 
             if self._run_mode == 'step' and not skip_first:
@@ -485,9 +488,6 @@ def _optim_update_params(self):
         else:
             self._optimizer.step()
 
-        if self._lr_scheduler is not None and self._lr_scheduler_mode == 'step':
-            self._lr_scheduler.step()
-
     @paddle.no_grad()
     def evaluate(self, epoch=1, valid_data_loader=None):
         """
@@ -724,12 +724,11 @@ def inference(self, data):
             tensorrt_config = None
             if 'TensorRT' in self._inference_configs:
                 tensorrt_config = TensorRTConfig(
-                                    **self._inference_configs['TensorRT'])
+                    **self._inference_configs['TensorRT'])
 
             self._inference_engine = InferenceEngine(
                 self._inference_configs['model_dir'],
-                self._inference_configs['mp_degree'],
-                tensorrt_config)
+                self._inference_configs['mp_degree'], tensorrt_config)
 
         return self._inference_engine.predict(data)
 

ppfleetx/data/transforms/preprocess.py

@@ -17,7 +17,6 @@
 from __future__ import unicode_literals
 
 from functools import partial
-import six
 import math
 import random
 import cv2
@@ -46,12 +45,8 @@ def __init__(self, to_rgb=True, channel_first=False):
         self.channel_first = channel_first
 
     def __call__(self, img):
-        if six.PY2:
-            assert type(img) is str and len(
-                img) > 0, "invalid input 'img' in DecodeImage"
-        else:
-            assert type(img) is bytes and len(
-                img) > 0, "invalid input 'img' in DecodeImage"
+        assert type(img) is bytes and len(
+            img) > 0, "invalid input 'img' in DecodeImage"
         data = np.frombuffer(img, dtype='uint8')
         img = cv2.imdecode(data, 1)
         if self.to_rgb:

ppfleetx/models/language_model/ernie/dygraph/hybrid_model.py

@@ -16,7 +16,6 @@
 import io
 import copy
 import logging
-import six
 import json
 
 import paddle

ppfleetx/models/language_model/ernie/dygraph/single_model.py

@@ -16,7 +16,6 @@
 import io
 import copy
 import logging
-import six
 import json
 
 import paddle