From c22cab83ddce48043449dc04da885d579290106a Mon Sep 17 00:00:00 2001 From: chaofengc Date: Wed, 26 Oct 2022 16:51:32 +0800 Subject: [PATCH] doc: print precalculated stat info in fid --- pyiqa/archs/fid_arch.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pyiqa/archs/fid_arch.py b/pyiqa/archs/fid_arch.py index 33b2391..9edc6a5 100644 --- a/pyiqa/archs/fid_arch.py +++ b/pyiqa/archs/fid_arch.py @@ -250,8 +250,8 @@ def forward(self, # compute fid of a folder elif fdir1 is not None and fdir2 is None: - if not verbose: - print(f"compute FID of a folder with {dataset_name} statistics") + if verbose: + print(f"compute FID of a folder with {dataset_name}-{dataset_res}-{dataset_split} statistics") fbname1 = os.path.basename(fdir1) np_feats1 = get_folder_features(fdir1, self.model, num_workers=num_workers, batch_size=batch_size, device=device, mode=mode, description=f"FID {fbname1}: ", verbose=verbose)