Update distill.py

vivekh2000 · web-flow · commit 166c902fe552 · 2024-06-11T22:50:17.000+05:30
This was redundant, not used anywhere.
diff --git 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
 

Original file line number	Diff line number	Diff line change
`@@ -127,7 +127,7 @@ def __init__(`
`127`	`127`	`)`
`128`	`128`
`129`	`129`	`def forward(self, img, labels, temperature = None, alpha = None, **kwargs):`
`130`		`- b, *_ = img.shape`
	`130`	`+`
`131`	`131`	`alpha = alpha if exists(alpha) else self.alpha`
`132`	`132`	`T = temperature if exists(temperature) else self.temperature`
`133`	`133`