Skip to content

Commit

Permalink
adding kwargs to sr-ste
Browse files Browse the repository at this point in the history
  • Loading branch information
mklasby committed Nov 1, 2024
1 parent 5cfa801 commit 3d3775e
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions sparsimony/pruners/ste.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,6 +28,7 @@ def __init__(
m: int = 4,
decay: None | float = 2e-4,
defaults: Optional[Dict[str, Any]] = None,
**kwargs,
):
self.scheduler = scheduler
self.distribution = distribution
Expand Down

0 comments on commit 3d3775e

Please sign in to comment.