Skip to content

Commit 22142e6

Browse files
committed
fix unknow keyword error caused by drop_rate keyword being sent by timm
1 parent 0842093 commit 22142e6

File tree

1 file changed

+1
-1
lines changed
  • ImageNet/training_scripts/imagenet_training/timm/models

1 file changed

+1
-1
lines changed

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -436,7 +436,7 @@ def remove_network_settings(kwargs: Dict[str, Any]) -> Dict[str, Any]:
436436
Returns:
437437
Dict[str,Any]: cleaned kwargs
438438
"""
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"]}
440440
return model_args
441441

442442

0 commit comments

Comments
 (0)