no weight scaler

jkobject · Jul 9, 2024 · 36c7b37 · 36c7b37
1 parent b343d2a
commit 36c7b37
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scdataloader/datamodule.py b/scdataloader/datamodule.py
@@ -233,7 +233,7 @@ def setup(self, stage=None):
             stage (str, optional): The stage of the model training process.
             It can be either 'fit' or 'test'. Defaults to None.
         """
-        if len(self.clss_to_weight) > 0:
+        if len(self.clss_to_weight) > 0 and self.weight_scaler > 0:
             weights, labels = self.dataset.get_label_weights(
                 self.clss_to_weight, scaler=self.weight_scaler
             )