Skip to content

Commit cc07dca

Browse files
david-waterworthDavid Waterworth
andauthored
corrected example usage of save_hyperparameters from List[str] to seperate str (Lightning-AI#2353)
Co-authored-by: David Waterworth <david.waterworth@cim.io>
1 parent aab9e77 commit cc07dca

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

docs/source/hyperparameters.rst

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -126,7 +126,7 @@ makes those values available via `self.hparams`.
126126
self.save_hyperparameters()
127127
128128
# equivalent
129-
self.save_hyperparameters(['layer_1_dim', 'learning_rate'])
129+
self.save_hyperparameters('layer_1_dim', 'learning_rate')
130130
131131
# this now works
132132
self.hparams.layer_1_dim
@@ -145,7 +145,7 @@ In that case, choose only a few
145145
self.loss_fx = loss_fx
146146
147147
# call this to save (layer_1_dim=128) to the checkpoint
148-
self.save_hyperparameters(['layer_1_dim'])
148+
self.save_hyperparameters('layer_1_dim')
149149
150150
# to load specify the other args
151151
model = LitMNIST.load_from_checkpoint(PATH, loss_fx=torch.nn.SomeOtherLoss, generator_network=MyGenerator())

0 commit comments

Comments
 (0)