Fix verbose warnings.

open-mmlab · Sep 8, 2021 · 9ba3c96 · 9ba3c96
1 parent 7147865
commit 9ba3c96
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mmcls/models/classifiers/image.py b/mmcls/models/classifiers/image.py
@@ -6,6 +6,8 @@
 from ..utils.augment import Augments
 from .base import BaseClassifier
 
+warnings.simplefilter('once')
+
 
 @CLASSIFIERS.register_module()
 class ImageClassifier(BaseClassifier):
@@ -74,7 +76,6 @@ def extract_feat(self, img):
         if self.return_tuple:
             if not isinstance(x, tuple):
                 x = (x, )
-                warnings.simplefilter('once')
                 warnings.warn(
                     'We will force all backbones to return a tuple in the '
                     'future. Please check your backbone and wrap the output '