From 166c902fe552832e38df07f6ca345591ccfb7e0e Mon Sep 17 00:00:00 2001
From: vivekh2000 <90518409+vivekh2000@users.noreply.github.com>
Date: Tue, 11 Jun 2024 22:50:17 +0530
Subject: [PATCH] Update distill.py

This was redundant, not used anywhere.
---
 vit_pytorch/distill.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/vit_pytorch/distill.py b/vit_pytorch/distill.py
index 79bf8c54..60032a91 100644
--- a/vit_pytorch/distill.py
+++ b/vit_pytorch/distill.py
@@ -127,7 +127,7 @@ def __init__(
         )
 
     def forward(self, img, labels, temperature = None, alpha = None, **kwargs):
-        b, *_ = img.shape
+        
         alpha = alpha if exists(alpha) else self.alpha
         T = temperature if exists(temperature) else self.temperature