feat: 🔧 add assertion information to fid

chaofengc · Oct 20, 2023 · c36d31c · c36d31c
1 parent 51a6fba
commit c36d31c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyiqa/archs/fid_arch.py b/pyiqa/archs/fid_arch.py
@@ -301,7 +301,7 @@ def forward(self,
 
         # compute fid of a folder
         elif fdir1 is not None and fdir2 is None:
-            assert dataset_name is not None, "The reference dataset_name should be specified to calcualte fid score."
+            assert dataset_name is not None, "When fdir2 is not provided, the reference dataset_name should be specified to calcualte fid score."
             if verbose:
                 print(f"compute FID of a folder with {dataset_name}-{mode}-{dataset_split}-{dataset_res} statistics")
             fbname1 = os.path.basename(fdir1)