fix bug

mmgen/core/evaluation/metrics.py

@@ -398,31 +398,38 @@ def feed(self, batch, mode):
                 return 0
 
             if isinstance(batch, dict):
-                batch_size = [v for v in batch.values()][0].shape[0] * ws
+                batch_size = [v for v in batch.values()][0].shape[0]
                 end = min(batch_size,
                           self.num_real_need - self.num_real_feeded)
                 batch_to_feed = {k: v[:end, ...] for k, v in batch.items()}
             else:
-                batch_size = batch.shape[0] * ws
+                batch_size = batch.shape[0]
                 end = min(batch_size,
                           self.num_real_need - self.num_real_feeded)
                 batch_to_feed = batch[:end, ...]
+
+            #
+            total_end = min(batch * ws,
+                            self.num_real_need - self.num_real_feeded)
             self.feed_op(batch_to_feed, mode)
-            self.num_real_feeded += end
+            self.num_real_feeded += total_end
             return end
 
         elif mode == 'fakes':
             if self.num_fake_feeded == self.num_fake_need:
                 return 0
 
-            batch_size = batch.shape[0] * ws
+            batch_size = batch.shape[0]
             end = min(batch_size, self.num_fake_need - self.num_fake_feeded)
             if isinstance(batch, dict):
                 batch_to_feed = {k: v[:end, ...] for k, v in batch.items()}
             else:
                 batch_to_feed = batch[:end, ...]
+
+            total_end = min(batch * ws,
+                            self.num_real_need - self.num_real_feeded)
             self.feed_op(batch_to_feed, mode)
-            self.num_fake_feeded += end
+            self.num_fake_feeded += total_end
             return end
         else:
             raise ValueError(