had loading weights in debuggmode

jcapp4 · jcapp4 · commit b6b668be62fc · 2025-10-15T18:59:05.000+02:00
diff --git a/chebai/preprocessing/datasets/base.py b/chebai/preprocessing/datasets/base.py
@@ -1185,7 +1185,7 @@ def load_processed_data(
                 data = data_df.to_dict(orient="records")
                 if kind == "train":
                     f.init_weights()
-                    data = f.add_val_weights(data)
+                    data = f.add_train_weights(data)
                 if kind == "validation":
                     data = f.add_val_weights(data)
                 # torch.save(data,"gewicht.pt")
