num gen corrected

pkhungurn · Jun 18, 2022 · e1bc9c3 · e1bc9c3
1 parent 44979df
commit e1bc9c3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/.gitignore b/.gitignore
@@ -23,6 +23,7 @@ figures/*
 
 setup.cfg
 examples/*
+notebooks/*
 
 tests/*
 scripts/tmp/*

diff --git a/cleanfid/fid.py b/cleanfid/fid.py
@@ -222,7 +222,7 @@ def get_model_features(G, model, mode="clean", z_dim=512,
                 resized_batch = img_batch
             feat = get_batch_features(resized_batch, model, device)
         l_feats.append(feat)
-    np_feats = np.concatenate(l_feats)
+    np_feats = np.concatenate(l_feats)[:num_gen]
     if return_z:
         latents = torch.cat(latents, 0)
         return np_feats, latents