From 35eb66c320b65a99379214db1802f51e593068e9 Mon Sep 17 00:00:00 2001
From: Mark Ma <519329064@qq.com>
Date: Tue, 1 Sep 2020 17:50:11 +0800
Subject: [PATCH] add progress bar display

---
 ppgan/metric/compute_fid.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/ppgan/metric/compute_fid.py b/ppgan/metric/compute_fid.py
index 8d000a0..c7f8c0a 100644
--- a/ppgan/metric/compute_fid.py
+++ b/ppgan/metric/compute_fid.py
@@ -23,9 +23,11 @@ import paddle.fluid as fluid
 from inception import InceptionV3
 from paddle.fluid.dygraph.base import to_variable
 
-
-def tqdm(x):
-    return x
+try:
+    from tqdm import tqdm
+except:
+    def tqdm(x):
+        return x
 
 
 """ based on https://github.com/mit-han-lab/gan-compression/blob/master/metric/fid_score.py
-- 
GitLab