fix tqdm

requirements.txt

@@ -1,10 +1,10 @@
+pillow>=7.2.0
 numpy>=1.16
 tqdm>=4.35.0
 torch>=1.3.0
 opencv-python>=4.5.2
 scikit-image>=0.14.0
 torchvision>=0.2.1
-pillow>=7.2.0
 lpips>=0.1.3
 gdown
 tqdm

train.py

@@ -114,6 +114,7 @@ def infer(args, humanflowmodel, image_names_list):
         if args.local_rank == 0:
             pbar.set_description(f"Epoch {i}/{train_num}")
             pbar.set_postfix({"data_time": data_time_interval, "train_time":train_time_interval})
+            pbar.update(1)
 
         with torch.no_grad():