Remove legacy compatibility for ViT_H_14 model

YosuaMichael · datumbox · web-flow · commit 9f603d64dcf1 · 2022-04-04T10:36:46.000+01:00
Co-authored-by: Vasilis Vryniotis &lt;datumbox@users.noreply.github.com&gt;
diff --git a/torchvision/models/vision_transformer.py b/torchvision/models/vision_transformer.py
@@ -554,7 +554,6 @@ def vit_l_32(*, weights: Optional[ViT_L_32_Weights] = None, progress: bool = Tru
     )
 
 
-@handle_legacy_interface(weights=("pretrained", ViT_H_14_Weights.IMAGENET1K_SWAG_V1))
 def vit_h_14(*, weights: Optional[ViT_H_14_Weights] = None, progress: bool = True, **kwargs: Any) -> VisionTransformer:
     """
     Constructs a vit_h_14 architecture from

Original file line number	Diff line number	Diff line change
`@@ -554,7 +554,6 @@ def vit_l_32(*, weights: Optional[ViT_L_32_Weights] = None, progress: bool = Tru`
`554`	`554`	`)`
`555`	`555`
`556`	`556`
`557`		`-@handle_legacy_interface(weights=("pretrained", ViT_H_14_Weights.IMAGENET1K_SWAG_V1))`
`558`	`557`	`def vit_h_14(, weights: Optional[ViT_H_14_Weights] = None, progress: bool = True, *kwargs: Any) -> VisionTransformer:`
`559`	`558`	`"""`
`560`	`559`	`Constructs a vit_h_14 architecture from`