Unverified Commit ec881a09 authored by Nicolas Hug's avatar Nicolas Hug Committed by GitHub
Browse files

Fix weight download URLs (#7898)

parent 655ebdbc
...@@ -463,7 +463,7 @@ class EfficientNet_B0_Weights(WeightsEnum): ...@@ -463,7 +463,7 @@ class EfficientNet_B0_Weights(WeightsEnum):
class EfficientNet_B1_Weights(WeightsEnum): class EfficientNet_B1_Weights(WeightsEnum):
IMAGENET1K_V1 = Weights( IMAGENET1K_V1 = Weights(
# Weights ported from https://github.com/rwightman/pytorch-image-models/ # Weights ported from https://github.com/rwightman/pytorch-image-models/
url="https://download.pytorch.org/models/efficientnet_b1_rwightman-533bc792.pth", url="https://download.pytorch.org/models/efficientnet_b1_rwightman-bac287d4.pth",
transforms=partial( transforms=partial(
ImageClassification, crop_size=240, resize_size=256, interpolation=InterpolationMode.BICUBIC ImageClassification, crop_size=240, resize_size=256, interpolation=InterpolationMode.BICUBIC
), ),
......
...@@ -108,7 +108,7 @@ class QuantizableGoogLeNet(GoogLeNet): ...@@ -108,7 +108,7 @@ class QuantizableGoogLeNet(GoogLeNet):
class GoogLeNet_QuantizedWeights(WeightsEnum): class GoogLeNet_QuantizedWeights(WeightsEnum):
IMAGENET1K_FBGEMM_V1 = Weights( IMAGENET1K_FBGEMM_V1 = Weights(
url="https://download.pytorch.org/models/quantized/googlenet_fbgemm-c00238cf.pth", url="https://download.pytorch.org/models/quantized/googlenet_fbgemm-c81f6644.pth",
transforms=partial(ImageClassification, crop_size=224), transforms=partial(ImageClassification, crop_size=224),
meta={ meta={
"num_params": 6624904, "num_params": 6624904,
......
...@@ -168,7 +168,7 @@ class QuantizableInception3(inception_module.Inception3): ...@@ -168,7 +168,7 @@ class QuantizableInception3(inception_module.Inception3):
class Inception_V3_QuantizedWeights(WeightsEnum): class Inception_V3_QuantizedWeights(WeightsEnum):
IMAGENET1K_FBGEMM_V1 = Weights( IMAGENET1K_FBGEMM_V1 = Weights(
url="https://download.pytorch.org/models/quantized/inception_v3_google_fbgemm-71447a44.pth", url="https://download.pytorch.org/models/quantized/inception_v3_google_fbgemm-a2837893.pth",
transforms=partial(ImageClassification, crop_size=299, resize_size=342), transforms=partial(ImageClassification, crop_size=299, resize_size=342),
meta={ meta={
"num_params": 27161264, "num_params": 27161264,
......
...@@ -148,7 +148,7 @@ class ShuffleNet_V2_X0_5_QuantizedWeights(WeightsEnum): ...@@ -148,7 +148,7 @@ class ShuffleNet_V2_X0_5_QuantizedWeights(WeightsEnum):
class ShuffleNet_V2_X1_0_QuantizedWeights(WeightsEnum): class ShuffleNet_V2_X1_0_QuantizedWeights(WeightsEnum):
IMAGENET1K_FBGEMM_V1 = Weights( IMAGENET1K_FBGEMM_V1 = Weights(
url="https://download.pytorch.org/models/quantized/shufflenetv2_x1_fbgemm-db332c57.pth", url="https://download.pytorch.org/models/quantized/shufflenetv2_x1_fbgemm-1e62bb32.pth",
transforms=partial(ImageClassification, crop_size=224), transforms=partial(ImageClassification, crop_size=224),
meta={ meta={
**_COMMON_META, **_COMMON_META,
......
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