Skip to content

Commit a3616ec

Browse files
committed
fix 9m config in cfg in simplenet models
1 parent 22142e6 commit a3616ec

File tree

3 files changed

+8
-8
lines changed

3 files changed

+8
-8
lines changed

Cifar/models/simplenet.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -80,10 +80,10 @@ def _cfg(url="", **kwargs):
8080
"simplenetv1_5m_m2": _cfg(
8181
url="https://github.com/Coderx7/SimpleNet_Pytorch/releases/download/v1.0.0/simplenetv1_5m_m2-c35297bf.pth"
8282
),
83-
"simplenetv1_m1_9m": _cfg(
83+
"simplenetv1_9m_m1": _cfg(
8484
url="https://github.com/Coderx7/SimpleNet_Pytorch/releases/download/v1.0.0/simplenetv1_9m_m1-8c98a0a5.pth"
8585
),
86-
"simplenetv1_m2_9m": _cfg(
86+
"simplenetv1_9m_m2": _cfg(
8787
url="https://github.com/Coderx7/SimpleNet_Pytorch/releases/download/v1.0.0/simplenetv1_9m_m2-6b01be1e.pth"
8888
),
8989
}
@@ -407,7 +407,7 @@ def remove_network_settings(kwargs: Dict[str, Any]) -> Dict[str, Any]:
407407
Returns:
408408
Dict[str,Any]: cleaned kwargs
409409
"""
410-
model_args = {k: v for k, v in kwargs.items() if k not in ["scale", "network_idx", "mode"]}
410+
model_args = {k: v for k, v in kwargs.items() if k not in ["scale", "network_idx", "mode", "drop_rate"]}
411411
return model_args
412412

413413

ImageNet/simplenet.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -80,10 +80,10 @@ def _cfg(url="", **kwargs):
8080
"simplenetv1_5m_m2": _cfg(
8181
url="https://github.com/Coderx7/SimpleNet_Pytorch/releases/download/v1.0.0/simplenetv1_5m_m2-c35297bf.pth"
8282
),
83-
"simplenetv1_m1_9m": _cfg(
83+
"simplenetv1_9m_m1": _cfg(
8484
url="https://github.com/Coderx7/SimpleNet_Pytorch/releases/download/v1.0.0/simplenetv1_9m_m1-8c98a0a5.pth"
8585
),
86-
"simplenetv1_m2_9m": _cfg(
86+
"simplenetv1_9m_m2": _cfg(
8787
url="https://github.com/Coderx7/SimpleNet_Pytorch/releases/download/v1.0.0/simplenetv1_9m_m2-6b01be1e.pth"
8888
),
8989
}
@@ -407,7 +407,7 @@ def remove_network_settings(kwargs: Dict[str, Any]) -> Dict[str, Any]:
407407
Returns:
408408
Dict[str,Any]: cleaned kwargs
409409
"""
410-
model_args = {k: v for k, v in kwargs.items() if k not in ["scale", "network_idx", "mode"]}
410+
model_args = {k: v for k, v in kwargs.items() if k not in ["scale", "network_idx", "mode", "drop_rate"]}
411411
return model_args
412412

413413

ImageNet/training_scripts/imagenet_training/timm/models/simplenet.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -76,10 +76,10 @@ def _cfg(url="", **kwargs):
7676
"simplenetv1_5m_m2": _cfg(
7777
url="https://github.com/Coderx7/SimpleNet_Pytorch/releases/download/v1.0.0/simplenetv1_5m_m2-c35297bf.pth"
7878
),
79-
"simplenetv1_m1_9m": _cfg(
79+
"simplenetv1_9m_m1": _cfg(
8080
url="https://github.com/Coderx7/SimpleNet_Pytorch/releases/download/v1.0.0/simplenetv1_9m_m1-8c98a0a5.pth"
8181
),
82-
"simplenetv1_m2_9m": _cfg(
82+
"simplenetv1_9m_m2": _cfg(
8383
url="https://github.com/Coderx7/SimpleNet_Pytorch/releases/download/v1.0.0/simplenetv1_9m_m2-6b01be1e.pth"
8484
),
8585
}

0 commit comments

Comments
 (0)