Skip to content

Commit 07cbb46

Browse files
authored
Update quantized shufflenet weights (#1715)
Previous weights are not compatible with current PyTorch
1 parent 300019d commit 07cbb46

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchvision/models/quantization/shufflenetv2.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
quant_model_urls = {
1616
'shufflenetv2_x0.5_fbgemm': None,
1717
'shufflenetv2_x1.0_fbgemm':
18-
'https://download.pytorch.org/models/quantized/shufflenetv2_x1_fbgemm-751f210b.pth',
18+
'https://download.pytorch.org/models/quantized/shufflenetv2_x1_fbgemm-db332c57.pth',
1919
'shufflenetv2_x1.5_fbgemm': None,
2020
'shufflenetv2_x2.0_fbgemm': None,
2121
}

0 commit comments

Comments
 (0)