Skip to content

Commit 1d54103

Browse files
committed
added ignore for hypter params
1 parent 025c30e commit 1d54103

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

src/lightning/pytorch/utilities/parsing.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -201,6 +201,9 @@ def save_hyperparameters(
201201
obj._hparams_name = "kwargs"
202202

203203
# `hparams` are expected here
204+
205+
# filter out hp based on ignore list
206+
hp = {k: v for k, v in hp.items() if k not in ignore}
204207
obj._set_hparams(hp)
205208

206209
for k, v in obj._hparams.items():

0 commit comments

Comments
 (0)