Skip to content

Commit 29bdc2e

Browse files
Update rnnlightregression.py
1 parent 5892799 commit 29bdc2e

1 file changed

Lines changed: 4 additions & 7 deletions

File tree

src/spotPython/light/regression/rnnlightregression.py

Lines changed: 4 additions & 7 deletions
Original file line numberDiff line numberDiff 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
)

0 commit comments

Comments
 (0)