From 1213eac51cf631ff36928d376f8a07a5d4d3b826 Mon Sep 17 00:00:00 2001
From: jiangjiajun <jiangjiajun@baidu.com>
Date: Tue, 12 May 2020 08:13:31 +0000
Subject: [PATCH] update log

---
 paddlex/cv/models/base.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/paddlex/cv/models/base.py b/paddlex/cv/models/base.py
index 112e3ef..923d0ec 100644
--- a/paddlex/cv/models/base.py
+++ b/paddlex/cv/models/base.py
@@ -199,7 +199,7 @@ class BaseAPI:
         self.exe.run(startup_prog)
         if pretrain_weights is not None:
             logging.info(
-                "Load pretrain weights from {}.".format(pretrain_weights))
+                "Load pretrain weights from {}.".format(pretrain_weights), use_color=True)
             paddlex.utils.utils.load_pretrain_weights(
                 self.exe, self.train_prog, pretrain_weights, fuse_bn)
         # 进行裁剪
@@ -211,7 +211,7 @@ class BaseAPI:
             from .slim.prune import get_params_ratios, prune_program
             logging.info(
                 "Start to prune program with eval_metric_loss = {}".format(
-                    eval_metric_loss))
+                    eval_metric_loss), use_color=True)
             origin_flops = paddleslim.analysis.flops(self.test_prog)
             prune_params_ratios = get_params_ratios(
                 sensitivities_file, eval_metric_loss=eval_metric_loss)
@@ -220,7 +220,7 @@ class BaseAPI:
             remaining_ratio = current_flops / origin_flops
             logging.info(
                 "Finish prune program, before FLOPs:{}, after prune FLOPs:{}, remaining ratio:{}"
-                .format(origin_flops, current_flops, remaining_ratio))
+                .format(origin_flops, current_flops, remaining_ratio), use_color=True)
             self.status = 'Prune'
 
     def get_model_info(self):
-- 
GitLab