Fix deploy error

2024-05-07 21:42:31 +08:00 · 2024-05-07 21:42:31 +08:00 · bcb31ca7ef
parent 17a886cb58
commit bcb31ca7ef
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: