kostyl-toolkit 0.1.27__py3-none-any.whl → 0.1.29__py3-none-any.whl

kostyl/ml/lightning/callbacks/registry_uploader.py

@@ -69,6 +69,8 @@ class ClearMLRegistryUploaderCallback(RegistryUploaderCallback):
         self._output_model: OutputModel | None = None
         self._last_uploaded_model_path: str = ""
         self._upload_callback: Callable | None = None
+
+        self._design_updated: bool = False
         return
 
     def _create_output_model(self) -> OutputModel:
@@ -116,7 +118,9 @@ class ClearMLRegistryUploaderCallback(RegistryUploaderCallback):
             auto_delete_file=False,
             async_enable=False,
         )
-        self._output_model.update_design(config_dict=self.config_dict)
+        if not self._design_updated and self.config_dict is not None:
+            self._output_model.update_design(config_dict=self.config_dict)
+            self._design_updated = True
 
         self._last_uploaded_model_path = path
         return

kostyl/ml/schedulers/linear.py

@@ -13,21 +13,21 @@ class _LinearScheduleBase(BaseScheduler):
         self,
         param_name: str,
         num_iters: int,
-        base_value: float,
+        start_value: float,
         final_value: float,
     ) -> None:
         self.param_name = param_name
         self.num_iters = num_iters
-        self.base_value = base_value
+        self.start_value = start_value
         self.final_value = final_value
 
         self.scheduler_values: npt.NDArray[np.float64] = np.array([], dtype=np.float64)
-        self.current_value_ = self.base_value
+        self.current_value_ = self.start_value
         return
 
     def _create_scheduler(self) -> None:
         self.scheduler_values = np.linspace(
-            self.base_value, self.final_value, num=self.num_iters, dtype=np.float64
+            self.start_value, self.final_value, num=self.num_iters, dtype=np.float64
         )
         if len(self.scheduler_values) != self.num_iters:
             raise ValueError(
@@ -69,7 +69,7 @@ class LinearScheduler(_LinearScheduleBase):
         optimizer: torch.optim.Optimizer,
         param_group_field: str,
         num_iters: int,
-        base_value: float,
+        start_value: float,
         final_value: float,
         multiplier_field: str | None = None,
         skip_if_zero: bool = False,
@@ -83,7 +83,7 @@ class LinearScheduler(_LinearScheduleBase):
             optimizer: Optimizer whose param groups are updated in-place.
             param_group_field: Name of the field that receives the scheduled value.
             num_iters: Number of scheduler iterations before clamping at ``final_value``.
-            base_value: Value used on the first iteration.
+            start_value: Value used on the first iteration.
             final_value: Value used once ``num_iters`` iterations are consumed.
             multiplier_field: Optional per-group multiplier applied to the scheduled value.
             skip_if_zero: Leave groups untouched when their target field equals zero.
@@ -99,7 +99,7 @@ class LinearScheduler(_LinearScheduleBase):
         super().__init__(
             param_name=param_group_field,
             num_iters=num_iters,
-            base_value=base_value,
+            start_value=start_value,
             final_value=final_value,
         )
         self.param_group_field = param_group_field

{kostyl_toolkit-0.1.27.dist-info → kostyl_toolkit-0.1.29.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: kostyl-toolkit
-Version: 0.1.27
+Version: 0.1.29
 Summary: Kickass Orchestration System for Training, Yielding & Logging 
 Requires-Dist: case-converter>=1.2.0
 Requires-Dist: loguru>=0.7.3

{kostyl_toolkit-0.1.27.dist-info → kostyl_toolkit-0.1.29.dist-info}/RECORD

@@ -13,7 +13,7 @@ kostyl/ml/lightning/__init__.py,sha256=-F3JAyq8KU1d-nACWryGu8d1CbvWbQ1rXFdeRwfE2
 kostyl/ml/lightning/callbacks/__init__.py,sha256=enexQt3octktsTiEYHltSF_24CM-NeFEVFimXiavGiY,296
 kostyl/ml/lightning/callbacks/checkpoint.py,sha256=KNwNVB2TFh2dcn133NbeTo5ul0jgiPYCeA-8NQ7U_mw,18951
 kostyl/ml/lightning/callbacks/early_stopping.py,sha256=D5nyjktCJ9XYAf28-kgXG8jORvXLl1N3nbDQnvValPM,615
-kostyl/ml/lightning/callbacks/registry_uploader.py,sha256=pIZHzHVANO_VsxPIbYhS8SwgZFHL341mP2HJnQ4iMFs,4216
+kostyl/ml/lightning/callbacks/registry_uploader.py,sha256=GtXKqjfBoqPa0udgIulnTrQZOK3NqUeSTEA9apjxz_o,4374
 kostyl/ml/lightning/extenstions/__init__.py,sha256=OY6QGv1agYgqqKf1xJBrxgp_i8FunVfPzYezfaRrGXU,182
 kostyl/ml/lightning/extenstions/custom_module.py,sha256=nB5jW7cqRD1tyh-q5LD2EtiFQwFkLXpnS9Yu6c5xMRg,5987
 kostyl/ml/lightning/extenstions/pretrained_model.py,sha256=QJGr2UvYJcU2Gy2w8z_cEvTodjv7hGdd2PPPfdOI-Mw,4017
@@ -26,11 +26,11 @@ kostyl/ml/schedulers/__init__.py,sha256=bxXbsU_WYnVbhvNNnuI7cOAh2Axz7D25TaleBTZh
 kostyl/ml/schedulers/base.py,sha256=9M2iOoOVSRojR_liPX1qo3Nn4iMXSM5ZJuAFWZTulUk,1327
 kostyl/ml/schedulers/composite.py,sha256=ee4xlMDMMtjKPkbTF2ue9GTr9DuGCGjZWf11mHbi6aE,2387
 kostyl/ml/schedulers/cosine.py,sha256=t74_ByT22L5NQKpnBVU9UGzBVx1ZM2GTylb9ct3_PVg,7627
-kostyl/ml/schedulers/linear.py,sha256=62mYEfd_2cQjOWrd0Vl5_sFeEokBKYmx496szhY04aU,5159
+kostyl/ml/schedulers/linear.py,sha256=7HPkVWcPa0lbaZywutXSDdVLLSihAyWk5XIE2Dzj_5Q,5168
 kostyl/utils/__init__.py,sha256=hkpmB6c5pr4Ti5BshOROebb7cvjDZfNCw83qZ_FFKMM,240
 kostyl/utils/dict_manipulations.py,sha256=e3vBicID74nYP8lHkVTQc4-IQwoJimrbFELy5uSF6Gk,1073
 kostyl/utils/fs.py,sha256=gAQNIU4R_2DhwjgzOS8BOMe0gZymtY1eZwmdgOdDgqo,510
 kostyl/utils/logging.py,sha256=Vye0u4-yeOSUc-f03gpQbxSktTbFiilTWLEVr00ZHvc,5796
-kostyl_toolkit-0.1.27.dist-info/WHEEL,sha256=ZyFSCYkV2BrxH6-HRVRg3R9Fo7MALzer9KiPYqNxSbo,79
-kostyl_toolkit-0.1.27.dist-info/METADATA,sha256=kg7Y2CJqhAI-3--rIKsPlarm1Ukk6jQLJpW2ZBvysI8,4269
-kostyl_toolkit-0.1.27.dist-info/RECORD,,
+kostyl_toolkit-0.1.29.dist-info/WHEEL,sha256=ZyFSCYkV2BrxH6-HRVRg3R9Fo7MALzer9KiPYqNxSbo,79
+kostyl_toolkit-0.1.29.dist-info/METADATA,sha256=x2GAeUoq_SaLJK1V48yxjw07IJYnhIvQ26bh8KE5LuQ,4269
+kostyl_toolkit-0.1.29.dist-info/RECORD,,