From 610d158e13e293b304a731c021daa444940722bc Mon Sep 17 00:00:00 2001 From: namsaraeva <a.namsaraeva@tum.de> Date: Wed, 29 May 2024 16:26:57 +0200 Subject: [PATCH] remove typo --- src/sparcscore/ml/plmodels.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/sparcscore/ml/plmodels.py b/src/sparcscore/ml/plmodels.py index 6c97da4..786f818 100644 --- a/src/sparcscore/ml/plmodels.py +++ b/src/sparcscore/ml/plmodels.py @@ -168,7 +168,7 @@ def test_step(self, batch, batch_idx): class RegressionModel(pl.LightningModule): - def __init__(self, model_type="VGG2_regression" **kwargs): + def __init__(self, model_type="VGG2_regression", **kwargs): super().__init__() self.save_hyperparameters()