radnn 0.1.0__tar.gz → 0.1.1__tar.gz

{radnn-0.1.0/src/radnn.egg-info → radnn-0.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: radnn
-Version: 0.1.0
+Version: 0.1.1
 Summary: Rapid Deep Neural Networks
 Author-email: "Pantelis I. Kaplanoglou" <pikaplanoglou@ihu.gr>
 License-Expression: MIT
@@ -27,4 +27,5 @@ Requires-Dist: matplotlib>=3.8.4
 Requires-Dist: pandas>=2.2.1
 Requires-Dist: scikit-learn>=1.4.2
 Requires-Dist: tqdm>=4.67.1
+Requires-Dist: psutil>=5.9.5
 Dynamic: license-file

{radnn-0.1.0 → radnn-0.1.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "radnn"
-version = "0.1.0"
+version = "0.1.1"
 description = "Rapid Deep Neural Networks"
 readme = "README.md"
 authors = [
@@ -29,7 +29,8 @@ dependencies = [
     "matplotlib >= 3.8.4",
     "pandas >= 2.2.1",
     "scikit-learn >= 1.4.2",
-	"tqdm >= 4.67.1"
+	"tqdm >= 4.67.1",
+	"psutil >= 5.9.5"
 ]
 
 [project.urls]

{radnn-0.1.0 → radnn-0.1.1}/src/radnn/__init__.py

@@ -5,9 +5,9 @@
 # Version 0.0.7.3  [2025-02-21]
 # Version 0.0.8    [2025-02-25]
 # Version 0.0.9    [2025-04-15]
-# Version 0.1.0    [2026-01-06]
-# Version 0.1.1    [2025-01-xx]
-__version__ = "0.1.0"
+# Version 0.1.0    [2026-01-07]
+# Version 0.1.1    [2025-01-08]
+__version__ = "0.1.1"
 
 from .system import FileStore, FileSystem
 from .ml_system import MLSystem

{radnn-0.1.0 → radnn-0.1.1}/src/radnn/learn/torch/ml_trainer.py

@@ -132,7 +132,7 @@ class MLModelTrainer():
       else:
         raise Exception(TRAINER_LR_SCHEDULER_INVALID_SETUP)
     
-    elif sSchedulingType.upper() == "CosineAnnealing":
+    elif sSchedulingType.upper() == "CosineAnnealing".upper():
       self.scheduler = torch.optim.lr_scheduler.CosineAnnealingLR(self.optimizer,
                                                                   T_max=nFinalChangeEpoch,
                                                                   eta_min=nFinalLR)
@@ -156,8 +156,10 @@ class MLModelTrainer():
     self.build_optimizer()
     self.build_lr_scheduler()
   # --------------------------------------------------------------------------------------------------------------------
-  def fit(self, device):
+  def fit(self, device, **kwargs):
     self.model.to(device)
+    bIsPreview = kwargs.get("is_preview", False)
+    
     hprm = self.hprm
     dInfo = {
               "experiment_name": hprm["Experiment.Name"],
@@ -203,7 +205,7 @@ class MLModelTrainer():
 
         self.optimizer.zero_grad()
         outputs = self.model(inputs)
-        loss = self.criterion(outputs, labels)
+        loss = self.criterion(outputs, labels).double()
         loss.backward()
         self.optimizer.step()
 
@@ -226,8 +228,8 @@ class MLModelTrainer():
         if not bInitialInfoSave:
           bInitialInfoSave = True
 
-        #if nDebugSteps == 4:
-          #break
+        if bIsPreview:
+          break
 
       nElapsedSecs = time.perf_counter() - nStart
       nStart = time.perf_counter()
@@ -292,7 +294,9 @@ class MLModelTrainer():
         epochs_without_improvement = 0
       else:
         epochs_without_improvement += 1
-
+      
+      if bIsPreview:
+        break
       nElapsedSecs = time.perf_counter() - nStart
 
       #if epochs_without_improvement >= patience:

{radnn-0.1.0 → radnn-0.1.1/src/radnn.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: radnn
-Version: 0.1.0
+Version: 0.1.1
 Summary: Rapid Deep Neural Networks
 Author-email: "Pantelis I. Kaplanoglou" <pikaplanoglou@ihu.gr>
 License-Expression: MIT
@@ -27,4 +27,5 @@ Requires-Dist: matplotlib>=3.8.4
 Requires-Dist: pandas>=2.2.1
 Requires-Dist: scikit-learn>=1.4.2
 Requires-Dist: tqdm>=4.67.1
+Requires-Dist: psutil>=5.9.5
 Dynamic: license-file

{radnn-0.1.0 → radnn-0.1.1}/src/radnn.egg-info/requires.txt

@@ -3,3 +3,4 @@ matplotlib>=3.8.4
 pandas>=2.2.1
 scikit-learn>=1.4.2
 tqdm>=4.67.1
+psutil>=5.9.5