Skip to content

Commit e56c4a0

Browse files
Update rnnlightregression.py
1 parent 88a5a0c commit e56c4a0

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

src/spotPython/light/regression/rnnlightregression.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,8 @@ class RNNLightRegression(L.LightningModule):
6363
lr_mult=0.1,
6464
patience=5,
6565
_L_in=10,
66-
_L_out=1)
66+
_L_out=1,
67+
_torchmetric="mean_squared_error")
6768
trainer = L.Trainer(max_epochs=2, enable_progress_bar=True)
6869
trainer.fit(net_light_base, train_loader)
6970
trainer.validate(net_light_base, val_loader)

0 commit comments

Comments
 (0)