Added pre-trained weights for ShuffleNetV2 x1.5 and x2.0 (#526)

* Added pre-trained weights for ShuffleNetV2 x1.5 and x2.0 * Update nanodet/model/backbone/shufflenetv2.py * Update nanodet/model/backbone/shufflenetv2.py --------- Co-authored-by: raember <[email protected]> Co-authored-by: RangiLyu <[email protected]>
RangiLyu · Aug 31, 2023 · 226d8cc · 226d8cc
1 parent 3c9607c
commit 226d8cc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nanodet/model/backbone/shufflenetv2.py b/nanodet/model/backbone/shufflenetv2.py
@@ -7,8 +7,8 @@
 model_urls = {
     "shufflenetv2_0.5x": "https://download.pytorch.org/models/shufflenetv2_x0.5-f707e7126e.pth",  # noqa: E501
     "shufflenetv2_1.0x": "https://download.pytorch.org/models/shufflenetv2_x1-5666bf0f80.pth",  # noqa: E501
-    "shufflenetv2_1.5x": None,
-    "shufflenetv2_2.0x": None,
+    "shufflenetv2_1.5x": "https://download.pytorch.org/models/shufflenetv2_x1_5-3c479a10.pth",  # noqa: E501
+    "shufflenetv2_2.0x": "https://download.pytorch.org/models/shufflenetv2_x2_0-8be3c8ee.pth",  # noqa: E501
 }