changes for lightning error

vidvath7 · vidvath7 · commit 43360bc700d5 · 2024-11-12T18:09:52.000+01:00
diff --git a/chebai/models/base.py b/chebai/models/base.py
@@ -46,7 +46,7 @@ def __init__(
         super().__init__()
         self.criterion = criterion
         self.save_hyperparameters(
-            ignore=["criterion", "train_metrics", "val_metrics", "test_metrics"]
+            ignore=["criterion", "train_metrics", "val_metrics", "test_metrics","_class_path"]
         )
         self.out_dim = out_dim
         if optimizer_kwargs:

Original file line number	Diff line number	Diff line change
`@@ -46,7 +46,7 @@ def __init__(`
`46`	`46`	`super().__init__()`
`47`	`47`	`self.criterion = criterion`
`48`	`48`	`self.save_hyperparameters(`
`49`		`- ignore=["criterion", "train_metrics", "val_metrics", "test_metrics"]`
	`49`	`+ ignore=["criterion", "train_metrics", "val_metrics", "test_metrics","_class_path"]`
`50`	`50`	`)`
`51`	`51`	`self.out_dim = out_dim`
`52`	`52`	`if optimizer_kwargs:`