[Fix] Force the sigmoid calculation in multi-label head to be full-pr…

…ecision. (#1251)

* [CI] Update CI of the master branch. (#1252)

* [Bot] Update assignee schedule. (#1262)

* [Fix] Force the sigmoid calculation in multi-label head to be
full-precision.

.circleci/config.yml

@@ -187,53 +187,23 @@ workflows:
             env: LRU_CACHE_CAPACITY=1
             requires:
               - lint
-        - build:
-            name: build_py36_torch1.6
-            torch: 1.6.0
-            torchvision: 0.7.0
-            requires:
-              - lint
         - build:
             name: build_py36_torch1.7
             torch: 1.7.0
             torchvision: 0.8.1
             requires:
               - lint
-        - build:
-            name: build_py36_torch1.8
-            torch: 1.8.0
-            torchvision: 0.9.0
-            requires:
-              - lint
-        - build:
-            name: build_py39_torch1.8
-            torch: 1.8.0
-            torchvision: 0.9.0
-            python: "3.9.0"
-            requires:
-              - lint
         - build:
             name: build_py39_torch1.9
             torch: 1.9.0
             torchvision: 0.10.0
             python: "3.9.0"
             requires:
               - lint
-        - build:
-            name: build_py39_torch1.10
-            torch: 1.10.0
-            torchvision: 0.11.1
-            python: "3.9.0"
-            requires:
-              - lint
         - build_with_cuda:
             name: build_py36_torch1.6_cu101
             requires:
               - build_with_timm
               - build_py36_torch1.5
-              - build_py36_torch1.6
               - build_py36_torch1.7
-              - build_py36_torch1.8
-              - build_py39_torch1.8
               - build_py39_torch1.9
-              - build_py39_torch1.10

.github/workflows/build.yml

@@ -35,7 +35,7 @@ jobs:
       UBUNTU_VERSION: ubuntu1804
     strategy:
       matrix:
-        python-version: [3.6]
+        python-version: [3.8]
         torch: [1.8.0]
         include:
           - torch: 1.8.0
@@ -84,43 +84,24 @@ jobs:
       UBUNTU_VERSION: ubuntu1804
     strategy:
       matrix:
-        python-version: [3.7]
-        torch: [1.5.0, 1.6.0, 1.7.0, 1.8.0, 1.9.0]
+        torch: [1.5.0, 1.8.0, 1.10.0, 1.13.0]
         include:
           - torch: 1.5.0
             torchvision: 0.6.0
             torch_major: 1.5.0
-          - torch: 1.6.0
-            torchvision: 0.7.0
-            torch_major: 1.6.0
-          - torch: 1.7.0
-            torchvision: 0.8.1
-            torch_major: 1.7.0
-          - torch: 1.8.0
-            torchvision: 0.9.0
-            torch_major: 1.8.0
-          - torch: 1.8.0
-            torchvision: 0.9.0
-            torch_major: 1.8.0
-            python-version: 3.8
+            python-version: '3.7'
           - torch: 1.8.0
             torchvision: 0.9.0
             torch_major: 1.8.0
-            python-version: 3.9
-          - torch: 1.9.0
-            torchvision: 0.10.0
-            torch_major: 1.9.0
-          - torch: 1.10.0
-            torchvision: 0.11.1
-            torch_major: 1.10.0
-          - torch: 1.10.0
-            torchvision: 0.11.1
-            torch_major: 1.10.0
-            python-version: 3.8
+            python-version: '3.8'
           - torch: 1.10.0
             torchvision: 0.11.1
             torch_major: 1.10.0
-            python-version: 3.9
+            python-version: '3.9'
+          - torch: 1.13.0
+            torchvision: 0.14.0
+            torch_major: 1.13.0
+            python-version: '3.10'
 
     steps:
       - uses: actions/checkout@v2

.owners.yml

@@ -0,0 +1,18 @@
+# assign issues to owners automatically
+assign:
+  issues: enabled # or disabled
+  pull_requests: enabled # or disabled
+  # assign strategy, both issues and pull requests follow the same strategy
+  strategy:
+    # random
+    # round-robin
+    daily-shift-based
+  schedule:
+    '*/1 * * * *'
+  # assignees
+  assignees:
+    - mzr1996
+    - tonysy
+    - Ezra-Yu
+    - techmonsterwang
+    - yingfhu

mmcls/models/heads/multi_label_head.py

@@ -82,7 +82,8 @@ def simple_test(self, x, sigmoid=True, post_process=True):
             x = x[-1]
 
         if sigmoid:
-            pred = torch.sigmoid(x) if x is not None else None
+            # Convert to full precision because sigmoid is sensitive.
+            pred = torch.sigmoid(x.float()) if x is not None else None
         else:
             pred = x
 

mmcls/models/heads/multi_label_linear_head.py

@@ -75,7 +75,9 @@ def simple_test(self, x, sigmoid=True, post_process=True):
         cls_score = self.fc(x)
 
         if sigmoid:
-            pred = torch.sigmoid(cls_score) if cls_score is not None else None
+            # Convert to full precision because sigmoid is sensitive.
+            pred = torch.sigmoid(
+                cls_score.float()) if cls_score is not None else None
         else:
             pred = cls_score