fix tqdm missing bug

zhangyujie05 · zhangyujie05 · commit 04760b0301ea · 2020-12-16T14:32:03.000+08:00
diff --git a/tools/general.py b/tools/general.py
@@ -13,6 +13,7 @@
 import time
 import torchvision
 from scipy.cluster.vq import kmeans
+from tqdm import tqdm
 
 @contextmanager
 def torch_distributed_zero_first(local_rank: int):
