feat: support ESNet

92ac78d1 · gaotingquan · ae6cb7d5 · 92ac78d1
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

paddleclas.py paddleclas.py +1 -0

未找到文件。
--- a/paddleclas.py
+++ b/paddleclas.py
@@ -65,6 +65,7 @@ MODEL_SERIES = {
        "EfficientNetB2", "EfficientNetB3", "EfficientNetB4", "EfficientNetB5",
        "EfficientNetB6", "EfficientNetB7"
    ],
+    "ESNet": ["ESNet_x0_25", "ESNet_x0_5", "ESNet_x0_75", "ESNet_x1_0"],
    "GhostNet":
    ["GhostNet_x0_5", "GhostNet_x1_0", "GhostNet_x1_3", "GhostNet_x1_3_ssld"],
    "HRNet": [