Fix

NicolasHug · NicolasHug · commit 6ddd1c2f2b79 · 2024-11-26T14:48:21.000Z
diff --git a/torchvision/models/maxvit.py b/torchvision/models/maxvit.py
@@ -688,7 +688,7 @@ def __init__(
                     p_stochastic=p_stochastic[p_idx : p_idx + num_layers],
                 ),
             )
-            input_size = self.blocks[-1].grid_size
+            input_size = self.blocks[-1].grid_size  # type: ignore[assignment]
             p_idx += num_layers
 
         # see https://github.com/google-research/maxvit/blob/da76cf0d8a6ec668cc31b399c4126186da7da944/maxvit/models/maxvit.py#L1137-L1158

Original file line number	Diff line number	Diff line change
`@@ -688,7 +688,7 @@ def __init__(`
`688`	`688`	`p_stochastic=p_stochastic[p_idx : p_idx + num_layers],`
`689`	`689`	`),`
`690`	`690`	`)`
`691`		`- input_size = self.blocks[-1].grid_size`
	`691`	`+ input_size = self.blocks[-1].grid_size # type: ignore[assignment]`
`692`	`692`	`p_idx += num_layers`
`693`	`693`
`694`	`694`	`# see https://github.com/google-research/maxvit/blob/da76cf0d8a6ec668cc31b399c4126186da7da944/maxvit/models/maxvit.py#L1137-L1158`