Skip to content

Commit c22cab8

Browse files
committed
doc: print precalculated stat info in fid
1 parent 1609c0e commit c22cab8

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

pyiqa/archs/fid_arch.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -250,8 +250,8 @@ def forward(self,
250250

251251
# compute fid of a folder
252252
elif fdir1 is not None and fdir2 is None:
253-
if not verbose:
254-
print(f"compute FID of a folder with {dataset_name} statistics")
253+
if verbose:
254+
print(f"compute FID of a folder with {dataset_name}-{dataset_res}-{dataset_split} statistics")
255255
fbname1 = os.path.basename(fdir1)
256256
np_feats1 = get_folder_features(fdir1, self.model, num_workers=num_workers, batch_size=batch_size,
257257
device=device, mode=mode, description=f"FID {fbname1}: ", verbose=verbose)

0 commit comments

Comments
 (0)