From 146201de8666248f8c2c3e1da7d9a67a6ff01479 Mon Sep 17 00:00:00 2001
From: MutableAI <mutableai@github.com>
Date: Wed, 31 May 2023 02:06:32 +0000
Subject: [PATCH] [code modification] test

---
 metrics.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/metrics.py b/metrics.py
index 37d6d39..97cb03e 100644
--- a/metrics.py
+++ b/metrics.py
@@ -18,7 +18,7 @@ def compute_fid_(model, images1, images2):
         features2 = model.latent_vector.cpu().numpy()
     
     means1 = features1.mean(0)    # (bs, num_features) --> (num_features)
-    means2 = features2.mean(0)
+    means2 = features2.mean(-1)
     
     # calculate mean and covariance statistics
     sigma1 = np.cov(features1, rowvar=False)