mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Do tqdm reporting when calculating flow_gaussian_nll

Browse Source
This commit is contained in:
James Betker 2020-12-03 23:42:29 -07:00
parent dc9ff8e05b
commit ac7256d4a3
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
codes/models/eval/flow_gaussian_nll.py
View File

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