radnn 0.1.0__py3-none-any.whl → 0.1.1__py3-none-any.whl

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/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.dist-info → radnn-0.1.1.dist-info}/METADATA

@@ -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.dist-info → radnn-0.1.1.dist-info}/RECORD

@@ -1,4 +1,4 @@
-radnn/__init__.py,sha256=9TBMAqk_yL3mM7GhAG8irLW7Eo9A3vIYRB5xrBmoLi0,489
+radnn/__init__.py,sha256=etsub-RbDx1I4hr_kKyzevw0TXIpE_pBQU5uIjR5ibs,489
 radnn/core.py,sha256=Q_JqMlytOubroTju27RuvS18NCCrC8Qp9vAHb3Vd0DA,10256
 radnn/errors.py,sha256=WZBFE08DtrSIaaAKfFpBLfJw_0banQkQcQybuY8zGvo,2245
 radnn/ml_system.py,sha256=E5GnpLNgK2iKLL3cOJ32M7jMUZRnAfXcJCAWFyzRJZ0,5874
@@ -50,7 +50,7 @@ radnn/learn/keras/keras_learning_rate_scheduler.py,sha256=ey1PPJS9ZThj0meUhsFKor
 radnn/learn/keras/keras_optimization_combo.py,sha256=gvkAR40xxhIKsQMXs_qLStINPskpxGwt_DE_97P8p-w,1799
 radnn/learn/torch/__init__.py,sha256=bgfVEVjSwy4tz0xGuj6VWc8euTMOcAomXXSGuDXGaMg,135
 radnn/learn/torch/ml_model_freezer.py,sha256=Y-uF0ckRvftFQqQaPDq1-e0wgSnpvq8BznzVdZWDeQk,9880
-radnn/learn/torch/ml_trainer.py,sha256=Y0ZwdQr35AlLqylkemnASy9bwtPKzP6fYrRnFfLcln0,21969
+radnn/learn/torch/ml_trainer.py,sha256=GbqJYcqUqAxornoDZDavnQNQiTt775ktwLxtvl29ZqQ,22087
 radnn/learn/torch/staircase_lr_scheduler.py,sha256=cV14U9pPVpxskzizR1euR5JJG_UcCxboXpF-i7fHIrI,613
 radnn/models/__init__.py,sha256=GTYp-CfYPvOmpUfqebJn9IiZRr5C-FU8oHpAYv4mhhM,207
 radnn/models/model_factory.py,sha256=8KV6zdi6DNoQ7pbOujUbCn_CIeFD2u6jGXwX3MOsGPI,4390
@@ -92,8 +92,8 @@ radnn/system/threads/thread_safe_string_collection.py,sha256=vdRMvwJ8CcLmsJ1uild
 radnn/system/threads/thread_worker.py,sha256=5KANBBHwnyaMvjyelBT1eyZCzRtH7MNZiHUhN1Xl1BY,3466
 radnn/test/__init__.py,sha256=XL9SgTJ6bGm3b0tcU3CroenP9rBm5XpDJozFGUv0UkQ,35
 radnn/test/tensor_hash.py,sha256=AbFUDZmcK76esZ28qx2rpDPyXuc7DtTPW9lIKQQy9zU,4200
-radnn-0.1.0.dist-info/licenses/LICENSE.txt,sha256=NMbnQdAQ2kWWQp2_8Sv2HG1p3jNvrPjYFSixnxPA3uE,1106
-radnn-0.1.0.dist-info/METADATA,sha256=Vyk3DVNwatIknnki9E3deT6GC0G0X3dFG9FGhEYLpho,1223
-radnn-0.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-radnn-0.1.0.dist-info/top_level.txt,sha256=FKlLIm6gRAeZlRzs-HCBJAB1q9ELJ7MgaL-qqFuPo6M,6
-radnn-0.1.0.dist-info/RECORD,,
+radnn-0.1.1.dist-info/licenses/LICENSE.txt,sha256=NMbnQdAQ2kWWQp2_8Sv2HG1p3jNvrPjYFSixnxPA3uE,1106
+radnn-0.1.1.dist-info/METADATA,sha256=5SwsvIdK4i6ScCaWkFWJMPyW4LTD-HF0rqlT4K2eAT8,1253
+radnn-0.1.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+radnn-0.1.1.dist-info/top_level.txt,sha256=FKlLIm6gRAeZlRzs-HCBJAB1q9ELJ7MgaL-qqFuPo6M,6
+radnn-0.1.1.dist-info/RECORD,,