Remove unused head argument

ar90n · Apr 6, 2022 · 9a1839a · 9a1839a
1 parent 9ff750e
commit 9a1839a
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/flash/image/embedding/strategies/default.py b/flash/image/embedding/strategies/default.py
@@ -28,11 +28,10 @@ class DefaultAdapter(Adapter):
 
     required_extras: str = "image"
 
-    def __init__(self, backbone: torch.nn.Module, head: torch.nn.Module):
+    def __init__(self, backbone: torch.nn.Module):
         super().__init__()
 
         self.backbone = backbone
-        self.head = head
 
     @classmethod
     @catch_url_error
@@ -41,10 +40,9 @@ def from_task(
         *args,
         task: AdapterTask,
         backbone: torch.nn.Module,
-        head: torch.nn.Module,
         **kwargs,
     ) -> Adapter:
-        adapter = cls(backbone, head)
+        adapter = cls(backbone)
         adapter.__dict__["_task"] = task
         return adapter