Fix performance issue with sample weights

See keras-team/keras#17357
MeteoSwiss · Mar 31, 2023 · 1b8ea15 · 1b8ea15
1 parent b3f8f0c
commit 1b8ea15
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mlpp_lib/train.py b/mlpp_lib/train.py
@@ -90,7 +90,7 @@ def train(
     res = model.fit(
         x=datamodule.train.x,
         y=datamodule.train.y,
-        sample_weight=datamodule.train.w,
+        sample_weight=(datamodule.train.w,) if datamodule.train.w is not None else None,
         epochs=cfg.get("epochs", 1),
         validation_data=(datamodule.val.x, datamodule.val.y),
         callbacks=callbacks,