We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 88a5a0c commit e56c4a0Copy full SHA for e56c4a0
1 file changed
src/spotPython/light/regression/rnnlightregression.py
@@ -63,7 +63,8 @@ class RNNLightRegression(L.LightningModule):
63
lr_mult=0.1,
64
patience=5,
65
_L_in=10,
66
- _L_out=1)
+ _L_out=1,
67
+ _torchmetric="mean_squared_error")
68
trainer = L.Trainer(max_epochs=2, enable_progress_bar=True)
69
trainer.fit(net_light_base, train_loader)
70
trainer.validate(net_light_base, val_loader)
0 commit comments