File tree Expand file tree Collapse file tree
src/spotPython/light/regression Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -124,9 +124,6 @@ def __init__(
124124 Returns:
125125 (NoneType): None
126126
127- Raises:
128- ValueError: If l1 is less than 4.
129-
130127 """
131128 super ().__init__ ()
132129 # Attribute 'act_fn' is an instance of `nn.Module` and is already saved during
@@ -142,9 +139,10 @@ def __init__(
142139 # set log_graph=True in Trainer to see the graph (in traintest.py)
143140 self .example_input_array = torch .zeros ((batch_size , self ._L_in ))
144141
145- # # Initialize RNN
146- # # input_size = number of features (= 11)
147- # # num_layers=1: only a single RNN and not stacked
142+ # Initialize RNN
143+ # input_size = number of features is defined via _L_in
144+ # output size via _L_out
145+ # num_layers=1: only a single RNN and not stacked
148146 rnn_units = self .hparams .l1
149147 fc_units = self .hparams .l1
150148
@@ -292,7 +290,6 @@ def configure_optimizers(self) -> torch.optim.Optimizer:
292290 torch.optim.Optimizer: The optimizer to use during training.
293291
294292 """
295- # optimizer = torch.optim.Adam(self.parameters(), lr=self.learning_rate)
296293 optimizer = optimizer_handler (
297294 optimizer_name = self .hparams .optimizer , params = self .parameters (), lr_mult = self .hparams .lr_mult
298295 )
You can’t perform that action at this time.
0 commit comments