add ad

.pre-commit-config.yaml

@@ -24,12 +24,6 @@
       files: \.(md|yml)$
     - id: check-case-conflict
 
-- repo: https://github.com/asottile/reorder_python_imports
-  rev: v1.9.0
-  hooks:
-  - id: reorder-python-imports
-    args: [--py3-plus]
-
 - repo: https://github.com/Lucas-C/pre-commit-hooks
   sha: v1.0.1
   hooks:

ppcls/utils/logger.py

@@ -72,10 +72,23 @@ def error(fmt, *args):
 
 
 def advertisement():
+    """
+    Show the advertising message like the following:
+
+    ===========================================================
+    ==        PaddleClas is powered by PaddlePaddle !        ==
+    ===========================================================
+    ==                                                       ==
+    ==   For more info please go to the following website.   ==
+    ==                                                       ==
+    ==       https://github.com/PaddlePaddle/PaddleClas      ==
+    ===========================================================
+
+    """
     copyright = "PaddleClas is powered by PaddlePaddle !"
     info = "For more info please go to the following website."
     website = "https://github.com/PaddlePaddle/PaddleClas"
-    AD_LEN = 55
+    AD_LEN = 6 + len(max([copyright, info, website], key=len))
 
     _logger.info("\n{0}\n{1}\n{2}\n{3}\n{4}\n{5}\n{6}\n{7}\n".format(
         "=" * (AD_LEN + 4),

tools/train.py

@@ -11,19 +11,23 @@
 # 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 argparse
 import os
 
 import paddle.fluid as fluid
-import program
 from paddle.fluid.incubate.fleet.base import role_maker
 from paddle.fluid.incubate.fleet.collective import fleet
 
 from ppcls.data import Reader
 from ppcls.utils import logger
 from ppcls.utils.config import get_config
-from ppcls.utils.save_load import init_model
-from ppcls.utils.save_load import save_model
+from ppcls.utils.save_load import init_model, save_model
+import program
 
 
 def parse_args():