Commit 22142e6

committed

fix unknow keyword error caused by drop_rate keyword being sent by timm

1 parent 0842093 commit 22142e6Copy full SHA for 22142e6

File tree

-1

lines changed

-1

lines changed

Lines changed: 1 addition & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -436,7 +436,7 @@ def remove_network_settings(kwargs: Dict[str, Any]) -> Dict[str, Any]:`
`436`	`436`	`Returns:`
`437`	`437`	`Dict[str,Any]: cleaned kwargs`
`438`	`438`	`"""`
`439`		`- model_args = {k: v for k, v in kwargs.items() if k not in ["scale", "network_idx", "mode"]}`
	`439`	`+ model_args = {k: v for k, v in kwargs.items() if k not in ["scale", "network_idx", "mode","drop_rate"]}`
`440`	`440`	`return model_args`
`441`	`441`
`442`	`442`

Comments

(0)