GANDLF 0.1.5.dev20250812__tar.gz → 0.1.5.dev20250813__tar.gz

{gandlf-0.1.5.dev20250812 → gandlf-0.1.5.dev20250813}/GANDLF/configuration/default_config.py

@@ -34,7 +34,7 @@ class DefaultParameters(BaseModel):
         default=10, description="Number of samples per volume."
     )
     q_num_workers: int = Field(
-        default=4, description="Number of worker threads to use."
+        default=0, description="Number of worker threads to use."
     )
     num_epochs: int = Field(default=100, description="Total number of epochs to train.")
     patience: int = Field(

{gandlf-0.1.5.dev20250812 → gandlf-0.1.5.dev20250813}/GANDLF/configuration/scheduler_config.py

@@ -84,7 +84,9 @@ schedulers_dict_config = {
     "reduce-on-plateau": ReduceOnPlateauConfig,
     "plateau": ReduceOnPlateauConfig,
     "reduceonplateau": ReduceOnPlateauConfig,
+    "cosineannealingwarmrestarts": CosineannealingConfig,
     "cosineannealing": CosineannealingConfig,
+    "cosineannealinglr": CosineannealingConfig,
     "warmupcosineschedule": WarmupcosinescheduleConfig,
     "wcs": WarmupcosinescheduleConfig,
 }

{gandlf-0.1.5.dev20250812 → gandlf-0.1.5.dev20250813}/GANDLF/schedulers/__init__.py

@@ -6,7 +6,8 @@ from .wrap_torch import (
     exp,
     step,
     reduce_on_plateau,
-    cosineannealing,
+    cosineannealingwarmrestarts,
+    cosineannealingLR,
 )
 
 from .wrap_monai import warmupcosineschedule
@@ -24,7 +25,9 @@ global_schedulers_dict = {
     "reduce-on-plateau": reduce_on_plateau,
     "plateau": reduce_on_plateau,
     "reduceonplateau": reduce_on_plateau,
-    "cosineannealing": cosineannealing,
+    "cosineannealing": cosineannealingwarmrestarts,
+    "cosineannealingwarmrestarts": cosineannealingwarmrestarts,
+    "cosineannealinglr": cosineannealingLR,
     "warmupcosineschedule": warmupcosineschedule,
     "wcs": warmupcosineschedule,
 }

{gandlf-0.1.5.dev20250812 → gandlf-0.1.5.dev20250813}/GANDLF/schedulers/wrap_torch.py

@@ -5,6 +5,7 @@ from torch.optim.lr_scheduler import (
     StepLR,
     ReduceLROnPlateau,
     CosineAnnealingWarmRestarts,
+    CosineAnnealingLR,
 )
 import math
 
@@ -169,14 +170,25 @@ def reduce_on_plateau(parameters):
     )
 
 
-def cosineannealing(parameters):
+def cosineannealingwarmrestarts(parameters):
     parameters["scheduler"]["T_0"] = parameters["scheduler"].get("T_0", 5)
     parameters["scheduler"]["T_mult"] = parameters["scheduler"].get("T_mult", 1)
-    parameters["scheduler"]["min_lr"] = parameters["scheduler"].get("min_lr", 0.001)
+    parameters["scheduler"]["eta_min"] = parameters["scheduler"].get("eta_min", 0.001)
 
     return CosineAnnealingWarmRestarts(
         parameters["optimizer_object"],
         T_0=parameters["scheduler"]["T_0"],
         T_mult=parameters["scheduler"]["T_mult"],
-        eta_min=parameters["scheduler"]["min_lr"],
+        eta_min=parameters["scheduler"]["eta_min"],
+    )
+
+
+def cosineannealingLR(parameters):
+    parameters["scheduler"]["T_max"] = parameters["scheduler"].get("T_max", 50)
+    parameters["scheduler"]["eta_min"] = parameters["scheduler"].get("eta_min", 0.001)
+
+    return CosineAnnealingLR(
+        parameters["optimizer_object"],
+        T_max=parameters["scheduler"]["T_max"],
+        eta_min=parameters["scheduler"]["eta_min"],
     )

{gandlf-0.1.5.dev20250812 → gandlf-0.1.5.dev20250813}/GANDLF/utils/tensor.py

@@ -290,7 +290,8 @@ def get_class_imbalance_weights_classification(
         np.fromiter(penalty_dict.values(), dtype=np.float64).sum()
         + sys.float_info.epsilon
     )
-    for i in range(params["model"]["num_classes"]):
+    # Only normalize penalties for classes that actually exist in penalty_dict
+    for i in penalty_dict.keys():
         penalty_dict[i] /= penalty_sum
 
     # passing None for sampling_weights because there is no clear way to calculate this for classification tasks which do not have a label

gandlf-0.1.5.dev20250813/GANDLF/version.py

@@ -0,0 +1 @@
+__version__ = "0.1.5-dev20250813"

{gandlf-0.1.5.dev20250812 → gandlf-0.1.5.dev20250813}/GANDLF.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: GANDLF
-Version: 0.1.5.dev20250812
+Version: 0.1.5.dev20250813
 Summary: PyTorch-based framework that handles segmentation/regression/classification using various DL architectures for medical imaging.
 Author: MLCommons
 Author-email: gandlf@mlcommons.org

{gandlf-0.1.5.dev20250812 → gandlf-0.1.5.dev20250813}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: GANDLF
-Version: 0.1.5.dev20250812
+Version: 0.1.5.dev20250813
 Summary: PyTorch-based framework that handles segmentation/regression/classification using various DL architectures for medical imaging.
 Author: MLCommons
 Author-email: gandlf@mlcommons.org

gandlf-0.1.5.dev20250812/GANDLF/version.py

@@ -1 +0,0 @@
-__version__ = "0.1.5-dev20250812"