diff --git a/codes/models/eval/flow_gaussian_nll.py b/codes/models/eval/flow_gaussian_nll.py index 2405a70d..f887b7c3 100644 --- a/codes/models/eval/flow_gaussian_nll.py +++ b/codes/models/eval/flow_gaussian_nll.py @@ -4,6 +4,7 @@ import torch import os.path as osp import torchvision from torch.utils.data import DataLoader +from tqdm import tqdm import models.eval.evaluator as evaluator from pytorch_fid import fid_score @@ -25,7 +26,8 @@ class FlowGaussianNll(evaluator.Evaluator): total_zs = 0 z_loss = 0 with torch.no_grad(): - for batch in self.dataloader: + print("Evaluating FlowGaussianNll..") + for batch in tqdm(self.dataloader): z, _, _ = self.model(gt=batch['GT'], lr=batch['LQ'], epses=[],