Merge bcb31ca7ef into ee7f2e8850

2024-11-14 09:09:29 +08:00 · 2024-11-14 09:09:29 +08:00 · 3de1a21793
parent ee7f2e8850 bcb31ca7ef
commit 3de1a21793
1 changed files with 1 additions and 1 deletions
--- a/mmpretrain/models/heads/levit_head.py
+++ b/mmpretrain/models/heads/levit_head.py
@ -59,7 +59,7 @@ class LeViTClsHead(ClsHead):
            self.head_dist = BatchNormLinear(in_channels, num_classes)

        if self.deploy:
-            self.switch_to_deploy(self)
+            self.switch_to_deploy()

    def switch_to_deploy(self):
        if self.deploy: