File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff 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())
You can’t perform that action at this time.
0 commit comments