Skip to content

GitLab

Unverified Commit a161098e authored by Vasilis Vryniotis's avatar Vasilis Vryniotis Committed by GitHub
Browse files

Update ShuffleNetV2 annotations for x1_5 and x2_0 variants (#6022)

parent 01664a8e
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
torchvision/models/shufflenetv2.py
View file @ a161098e
......@@ -316,7 +316,7 @@ def shufflenet_v2_x1_0(
return _shufflenetv2(weights, progress, [4, 8, 4], [24, 116, 232, 464, 1024], **kwargs)
@handle_legacy_interface(weights=("pretrained", None))
@handle_legacy_interface(weights=("pretrained", ShuffleNet_V2_X1_5_Weights.IMAGENET1K_V1))
def shufflenet_v2_x1_5(
*, weights: Optional[ShuffleNet_V2_X1_5_Weights] = None, progress: bool = True, **kwargs: Any
) -> ShuffleNetV2:
......@@ -346,7 +346,7 @@ def shufflenet_v2_x1_5(
return _shufflenetv2(weights, progress, [4, 8, 4], [24, 176, 352, 704, 1024], **kwargs)
@handle_legacy_interface(weights=("pretrained", None))
@handle_legacy_interface(weights=("pretrained", ShuffleNet_V2_X2_0_Weights.IMAGENET1K_V1))
def shufflenet_v2_x2_0(
*, weights: Optional[ShuffleNet_V2_X2_0_Weights] = None, progress: bool = True, **kwargs: Any
) -> ShuffleNetV2:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment