Commit a3616ec

committed

fix 9m config in cfg in simplenet models

1 parent 22142e6 commit a3616ecCopy full SHA for a3616ec

File tree

-8

lines changed

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

-8

lines changed

Lines changed: 3 additions & 3 deletions

Original file line number	Diff line number	Diff line change
`@@ -80,10 +80,10 @@ def _cfg(url="", **kwargs):`
`80`	`80`	`"simplenetv1_5m_m2": _cfg(`
`81`	`81`	`url="https://github.com/Coderx7/SimpleNet_Pytorch/releases/download/v1.0.0/simplenetv1_5m_m2-c35297bf.pth"`
`82`	`82`	`),`
`83`		`- "simplenetv1_m1_9m": _cfg(`
	`83`	`+ "simplenetv1_9m_m1": _cfg(`
`84`	`84`	`url="https://github.com/Coderx7/SimpleNet_Pytorch/releases/download/v1.0.0/simplenetv1_9m_m1-8c98a0a5.pth"`
`85`	`85`	`),`
`86`		`- "simplenetv1_m2_9m": _cfg(`
	`86`	`+ "simplenetv1_9m_m2": _cfg(`
`87`	`87`	`url="https://github.com/Coderx7/SimpleNet_Pytorch/releases/download/v1.0.0/simplenetv1_9m_m2-6b01be1e.pth"`
`88`	`88`	`),`
`89`	`89`	`}`
`@@ -407,7 +407,7 @@ def remove_network_settings(kwargs: Dict[str, Any]) -> Dict[str, Any]:`
`407`	`407`	`Returns:`
`408`	`408`	`Dict[str,Any]: cleaned kwargs`
`409`	`409`	`"""`
`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"]}`
`411`	`411`	`return model_args`
`412`	`412`
`413`	`413`

Lines changed: 3 additions & 3 deletions

Original file line number	Diff line number	Diff line change
`@@ -80,10 +80,10 @@ def _cfg(url="", **kwargs):`
`80`	`80`	`"simplenetv1_5m_m2": _cfg(`
`81`	`81`	`url="https://github.com/Coderx7/SimpleNet_Pytorch/releases/download/v1.0.0/simplenetv1_5m_m2-c35297bf.pth"`
`82`	`82`	`),`
`83`		`- "simplenetv1_m1_9m": _cfg(`
	`83`	`+ "simplenetv1_9m_m1": _cfg(`
`84`	`84`	`url="https://github.com/Coderx7/SimpleNet_Pytorch/releases/download/v1.0.0/simplenetv1_9m_m1-8c98a0a5.pth"`
`85`	`85`	`),`
`86`		`- "simplenetv1_m2_9m": _cfg(`
	`86`	`+ "simplenetv1_9m_m2": _cfg(`
`87`	`87`	`url="https://github.com/Coderx7/SimpleNet_Pytorch/releases/download/v1.0.0/simplenetv1_9m_m2-6b01be1e.pth"`
`88`	`88`	`),`
`89`	`89`	`}`
`@@ -407,7 +407,7 @@ def remove_network_settings(kwargs: Dict[str, Any]) -> Dict[str, Any]:`
`407`	`407`	`Returns:`
`408`	`408`	`Dict[str,Any]: cleaned kwargs`
`409`	`409`	`"""`
`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"]}`
`411`	`411`	`return model_args`
`412`	`412`
`413`	`413`

Lines changed: 2 additions & 2 deletions

Original file line number	Diff line number	Diff line change
`@@ -76,10 +76,10 @@ def _cfg(url="", **kwargs):`
`76`	`76`	`"simplenetv1_5m_m2": _cfg(`
`77`	`77`	`url="https://github.com/Coderx7/SimpleNet_Pytorch/releases/download/v1.0.0/simplenetv1_5m_m2-c35297bf.pth"`
`78`	`78`	`),`
`79`		`- "simplenetv1_m1_9m": _cfg(`
	`79`	`+ "simplenetv1_9m_m1": _cfg(`
`80`	`80`	`url="https://github.com/Coderx7/SimpleNet_Pytorch/releases/download/v1.0.0/simplenetv1_9m_m1-8c98a0a5.pth"`
`81`	`81`	`),`
`82`		`- "simplenetv1_m2_9m": _cfg(`
	`82`	`+ "simplenetv1_9m_m2": _cfg(`
`83`	`83`	`url="https://github.com/Coderx7/SimpleNet_Pytorch/releases/download/v1.0.0/simplenetv1_9m_m2-6b01be1e.pth"`
`84`	`84`	`),`
`85`	`85`	`}`

Comments

(0)