From bcb31ca7efc166366b13abfd8126ff018443698c Mon Sep 17 00:00:00 2001
From: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com>
Date: Tue, 7 May 2024 21:42:31 +0800
Subject: [PATCH] Fix deploy error

---
 mmpretrain/models/heads/levit_head.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mmpretrain/models/heads/levit_head.py b/mmpretrain/models/heads/levit_head.py
index a74d7ecc..46418179 100644
--- 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: