From 4ebee155e89e82a821f5588d0e3a68a13ab7437f Mon Sep 17 00:00:00 2001
From: Mingqiang Ning <ningmq@qq.com>
Date: Wed, 30 Jun 2021 07:57:21 -0500
Subject: [PATCH] fix a bug when samples_per_gpu==1 (#311)

---
 mmcls/models/classifiers/image.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/mmcls/models/classifiers/image.py b/mmcls/models/classifiers/image.py
index 043a9eb2..283d86a9 100644
--- a/mmcls/models/classifiers/image.py
+++ b/mmcls/models/classifiers/image.py
@@ -94,4 +94,7 @@ class ImageClassifier(BaseClassifier):
     def simple_test(self, img, img_metas):
         """Test without augmentation."""
         x = self.extract_feat(img)
+        x_dims = len(x.shape)
+        if x_dims == 1:
+            x.unsqueeze_(0)
         return self.head.simple_test(x)