dsipts 1.1.20__tar.gz → 1.1.21__tar.gz

{dsipts-1.1.20 → dsipts-1.1.21}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dsipts
-Version: 1.1.20
+Version: 1.1.21
 Summary: Unified library for timeseries modelling
 Author-email: Andrea Gobbi <agobbi@fbk.eu>
 Project-URL: Homepage, https://github.com/DSIP-FBK/DSIPTS

{dsipts-1.1.20 → dsipts-1.1.21}/pyproject.toml

@@ -1,7 +1,7 @@
 
 [project]
 name = "dsipts"
-version = "1.1.20"
+version = "1.1.21"
 description = "Unified library for timeseries modelling"
 readme = { file = "LONG_DESCRIPTION.md", content-type = "text/markdown" }
 requires-python = ">=3.12"
@@ -40,7 +40,7 @@ Homepage = "https://github.com/DSIP-FBK/DSIPTS"
 Docs = "https://dsip-fbk.github.io/DSIPTS/"
 
 [tool.bumpver]
-current_version = "1.1.20"
+current_version = "1.1.21"
 version_pattern = "MAJOR.MINOR.PATCH"
 commit = false
 tag = false

{dsipts-1.1.20 → dsipts-1.1.21}/src/dsipts/data_structure/data_structure.py

@@ -774,6 +774,7 @@ class TimeSeries():
             train_dl = DataLoader(train, batch_size = batch_size , shuffle=True,drop_last=True,num_workers=num_workers,persistent_workers=persistent_workers)
         valid_dl = DataLoader(validation, batch_size = batch_size , shuffle=False,drop_last=True,num_workers=num_workers,persistent_workers=persistent_workers)
         
+        beauty_string(f'train:{len(train_dl)}, validation:{len(valid_dl)}','section',self.verbose)
 
         if debug_prediction:
             dl = DataLoader(test, batch_size = batch_size , shuffle=False,drop_last=True,num_workers=num_workers,persistent_workers=persistent_workers)
@@ -801,6 +802,8 @@ class TimeSeries():
         
         
         #logger = CSVLogger("logs", name=dirpath)
+        beauty_string(f'Init aim','section',self.verbose)
+
         aim_logger = AimLogger(
             experiment=self.name,
             train_metric_prefix='train_',
@@ -884,7 +887,8 @@ class TimeSeries():
         tot_seconds = time.time()
 
 
-      
+        beauty_string(f'tuning lr','section',self.verbose)
+
      
         if auto_lr_find and (weight_exists is False):
             try:
@@ -902,7 +906,8 @@ class TimeSeries():
                     self.model.optim_config['lr'] = lr_finder.suggestion() ## we are using it as optim key
             except Exception as e:
                 beauty_string(f'There is a problem with the finding LR routine {e}','section',self.verbose)
-        
+        beauty_string(f'Start fitting','section',self.verbose)
+
         if OLD_PL:
             if weight_exists:
                 trainer.fit(self.model, train_dl,valid_dl,ckpt_path=os.path.join(dirpath,'last.ckpt'))

dsipts-1.1.21/src/dsipts/version.py

@@ -0,0 +1 @@
+__version__ = "1.1.21"

{dsipts-1.1.20 → dsipts-1.1.21}/src/dsipts.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dsipts
-Version: 1.1.20
+Version: 1.1.21
 Summary: Unified library for timeseries modelling
 Author-email: Andrea Gobbi <agobbi@fbk.eu>
 Project-URL: Homepage, https://github.com/DSIP-FBK/DSIPTS

dsipts-1.1.20/src/dsipts/version.py

@@ -1 +0,0 @@
-__version__ = "1.1.20"