polish code style

.pre-commit-config.yaml

@@ -3,9 +3,13 @@
   hooks:
     - id: yapf
       files: \.py$
+
 - repo: https://github.com/pre-commit/pre-commit-hooks
   sha: a11d9314b22d8f8c7556443875b731ef05965464
   hooks:
+    - id: flake8
+      args: ['--ignore=E265']
+    - id: check-yaml
     - id: check-merge-conflict
     - id: check-symlinks
     - id: detect-private-key
@@ -14,6 +18,7 @@
       files: \.(md|yml)$
     - id: trailing-whitespace
       files: \.(md|yml)$
+
 - repo: https://github.com/Lucas-C/pre-commit-hooks
   sha: v1.0.1
   hooks:

tools/eval.py

-#copyright (c) 2020 PaddlePaddle Authors. All Rights Reserve.
+# copyright (c) 2020 PaddlePaddle Authors. All Rights Reserve.
 #
-#Licensed under the Apache License, Version 2.0 (the "License");
-#you may not use this file except in compliance with the License.
-#You may obtain a copy of the License at
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
 #
 #    http://www.apache.org/licenses/LICENSE-2.0
 #
-#Unless required by applicable law or agreed to in writing, software
-#distributed under the License is distributed on an "AS IS" BASIS,
-#WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-#See the License for the specific language governing permissions and
-#limitations under the License.
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
 
 from __future__ import absolute_import
 from __future__ import division
 from __future__ import print_function
 
-import sys
+import os
 import argparse
 
-import paddle
 import paddle.fluid as fluid
 
 import program
 
 from ppcls.data import Reader
 from ppcls.utils.config import get_config
-from ppcls.utils.save_load import init_model, save_model
-from ppcls.utils import logger
+from ppcls.utils.save_load import init_model
 
 from paddle.fluid.incubate.fleet.collective import fleet
 from paddle.fluid.incubate.fleet.base import role_maker
@@ -39,7 +37,7 @@ def parse_args():
         '-c',
         '--config',
         type=str,
-        default='configs/eval.yaml',
+        default='./configs/eval.yaml',
         help='config file path')
     parser.add_argument(
         '-o',
@@ -69,7 +67,7 @@ def main(args):
     exe = fluid.Executor(place)
     exe.run(startup_prog)
 
-    init_model(config, valid_prog, exe, 'ppcls')
+    init_model(config, valid_prog, exe)
 
     valid_reader = Reader(config, 'valid')()
     valid_dataloader.set_sample_list_generator(valid_reader, place)

tools/train.py

-#copyright (c) 2020 PaddlePaddle Authors. All Rights Reserve.
+# copyright (c) 2020 PaddlePaddle Authors. All Rights Reserve.
 #
-#Licensed under the Apache License, Version 2.0 (the "License");
-#you may not use this file except in compliance with the License.
-#You may obtain a copy of the License at
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
 #
 #    http://www.apache.org/licenses/LICENSE-2.0
 #
-#Unless required by applicable law or agreed to in writing, software
-#distributed under the License is distributed on an "AS IS" BASIS,
-#WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-#See the License for the specific language governing permissions and
-#limitations under the License.
-
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
 
 from __future__ import absolute_import
 from __future__ import division
@@ -20,7 +19,6 @@ from __future__ import print_function
 import argparse
 import os
 
-import paddle
 import paddle.fluid as fluid
 
 import program
@@ -28,7 +26,6 @@ import program
 from ppcls.data import Reader
 from ppcls.utils.config import get_config
 from ppcls.utils.save_load import init_model, save_model
-from ppcls.utils import logger
 
 from paddle.fluid.incubate.fleet.collective import fleet
 from paddle.fluid.incubate.fleet.base import role_maker