dsipts 1.1.20__py3-none-any.whl → 1.1.21__py3-none-any.whl

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/version.py

@@ -1 +1 @@
-__version__ = "1.1.20"
+__version__ = "1.1.21"

{dsipts-1.1.20.dist-info → dsipts-1.1.21.dist-info}/METADATA

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

@@ -1,10 +1,10 @@
 dsipts/__init__.py,sha256=2KK6XbUewsjmQQ6-zNMKIcFP01jtcXskcwr3qZwQToE,1822
-dsipts/version.py,sha256=356mX5gt6SA0pAUpV03syJebLSqgcNO5cBqfmKn-49E,22
+dsipts/version.py,sha256=gTyF8-tCUVFi4ATGIeTj_wdg8dNApZafzTAEIzCCdAM,22
 dsipts/data_management/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 dsipts/data_management/monash.py,sha256=aZxq9FbIH6IsU8Lwou1hAokXjgOAK-wdl2VAeFg2k4M,13075
 dsipts/data_management/public_datasets.py,sha256=yXFzOZZ-X0ZG1DoqVU-zFmEGVMc2033YDQhRgYxY8ws,6793
 dsipts/data_structure/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-dsipts/data_structure/data_structure.py,sha256=co4AcBvDGxTQW4ibeURwudpyNjVwDMS8geoFxAgBY_Q,64929
+dsipts/data_structure/data_structure.py,sha256=-LLhUNs1ymAGCujU0WRXQ8ZS9kYgt1e5k__1lHfLads,65195
 dsipts/data_structure/modifiers.py,sha256=qlry9dfw8pEE0GrvgwROZJkJ6oPpUnjEHPIG5qIetss,7948
 dsipts/data_structure/utils.py,sha256=ZL-z_InmFUkge5kQoHSrev1t6nyve9sTYTVeA75Or-I,6689
 dsipts/models/Autoformer.py,sha256=nUQvPC_qtajLT1AHdNJmF_P3ZL01j3spkZ4ubxdGF3g,8497
@@ -83,7 +83,7 @@ dsipts/models/vva/minigpt.py,sha256=bg0JddqSD322uxSGexen3nPXL_hGTsk3vNLR62d7-w8,
 dsipts/models/vva/vqvae.py,sha256=RzCQ_M9xBprp7_x20dSV3EQqlO0FjPUGWV-qdyKrQsM,19680
 dsipts/models/xlstm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 dsipts/models/xlstm/xLSTM.py,sha256=ZKZZmffmIq1Vb71CR4GSyM8viqVx-u0FChxhcNgHub8,10081
-dsipts-1.1.20.dist-info/METADATA,sha256=q4EZto89qbSuK1A0SyrIXLAssEIZCKhvq7qaqPGcupY,27614
-dsipts-1.1.20.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-dsipts-1.1.20.dist-info/top_level.txt,sha256=i6o0rf5ScFwZK21E89dSKjVNjUBkrEQpn0-Vij43748,7
-dsipts-1.1.20.dist-info/RECORD,,
+dsipts-1.1.21.dist-info/METADATA,sha256=zyKSU0VS9u0rgMd6qGpr5NCrh2h27f85jaTQtYu2VPc,27614
+dsipts-1.1.21.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+dsipts-1.1.21.dist-info/top_level.txt,sha256=i6o0rf5ScFwZK21E89dSKjVNjUBkrEQpn0-Vij43748,7
+dsipts-1.1.21.dist-info/RECORD,,