spikezoo 0.2.3.3__py3-none-any.whl → 0.2.3.4__py3-none-any.whl

spikezoo/pipeline/base_pipeline.py

@@ -56,7 +56,7 @@ class Pipeline:
 
     def _setup_model_data(self, model_cfg, dataset_cfg):
         """Model and Data setup."""
-        self.logger.info("Model and dataset is setting up...")
+        print("Model and dataset is setting up...")
         # model [1] build the model. [2] build the network.
         self.model: BaseModel = build_model_name(model_cfg) if isinstance(model_cfg, str) else build_model_cfg(model_cfg)
         self.model.build_network(mode="eval", version=self.cfg.version)
@@ -69,7 +69,7 @@ class Pipeline:
 
     def _setup_pipeline(self):
         """Pipeline setup."""
-        self.logger.info("Pipeline is setting up...")
+        print("Pipeline is setting up...")
         # save folder
         self.thistime = datetime.now().strftime("%Y-%m-%d_%H-%M-%S-%f")[:23]
         self.save_folder = Path(f"results") if len(self.cfg.save_folder) == 0 else self.cfg.save_folder

spikezoo/pipeline/train_cfgs.py

@@ -30,28 +30,6 @@ class REDS_BASE_TrainConfig(TrainPipelineConfig):
     scheduler_cfg: Optional[SchedulerConfig] = MultiStepSchedulerConfig(milestones=[400], gamma=0.2) # from wgse
     loss_weight_dict: Dict = field(default_factory=lambda: {"l1": 1})
 
-# @dataclass
-# class REDS_BASE_TrainConfig(TrainPipelineConfig):
-#     """Training setting for methods on the REDS-BASE dataset."""
-
-#     # parameters setting
-#     epochs: int = 700
-#     steps_per_save_imgs: int = 200
-#     steps_per_save_ckpt: int = 500
-#     steps_per_cal_metrics: int = 100
-#     metric_names: List[str] = field(default_factory=lambda: ["psnr", "ssim"])
-
-#     # dataloader setting
-#     bs_train: int = 8
-#     num_workers: int = 4
-#     pin_memory: bool = False
-
-#     # train setting - optimizer & scheduler & loss_dict
-#     optimizer_cfg: OptimizerConfig = AdamOptimizerConfig(lr=1e-4)
-#     scheduler_cfg: Optional[SchedulerConfig] = MultiStepSchedulerConfig(milestones=[400, 600], gamma=0.2) # from wgse
-#     loss_weight_dict: Dict = field(default_factory=lambda: {"l1": 1})
-
-
 # ! Train Config for each method on the official setting, not recommended to utilize their default parameters owing to the dataset setting.
 @dataclass
 class BSFTrainConfig(TrainPipelineConfig):

{spikezoo-0.2.3.3.dist-info → spikezoo-0.2.3.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: spikezoo
-Version: 0.2.3.3
+Version: 0.2.3.4
 Summary: A deep learning toolbox for spike-to-image models.
 Home-page: https://github.com/chenkang455/Spike-Zoo
 Author: Kang Chen

{spikezoo-0.2.3.3.dist-info → spikezoo-0.2.3.4.dist-info}/RECORD

@@ -154,9 +154,9 @@ spikezoo/models/tfi_model.py,sha256=tgD_HsiXk9jGuh5f_Bh6c3BqJi1p5DWCVo4N1tp5fgs,
 spikezoo/models/tfp_model.py,sha256=ihl1H__bWIbE9oair_t8rNJ5qnPJPKl-r_DpaO-0Sdk,663
 spikezoo/models/wgse_model.py,sha256=Kl9uV-LeO0Lj7SuPQ9pglw1Khs2b-7miS3A_faL6WSU,805
 spikezoo/pipeline/__init__.py,sha256=WPsukNR4cannwsghiukqNsWbWGH5DVPapR_Ly-WOU4Q,188
-spikezoo/pipeline/base_pipeline.py,sha256=3laGM-cMhNTSfCXx5jY53V8GY5BhPhHK48yjqm-Gre0,13478
+spikezoo/pipeline/base_pipeline.py,sha256=y_jCmYlNlKnK-607b6p987tA3eDnrOI8S9XJGB0hwUE,13456
 spikezoo/pipeline/ensemble_pipeline.py,sha256=ljZkGiCCpxvpC04Aa-r_tvBnqcBpUVi9fl_878tJAcg,2555
-spikezoo/pipeline/train_cfgs.py,sha256=6NO7DfPc7yjJfOrcIPQPfUPbUODz6eRKurEIDjMmaxA,3836
+spikezoo/pipeline/train_cfgs.py,sha256=ZzTGKlAwkQGDsI0CBfT0qs6a_sVfSWJWJJgTEjQk7C8,3028
 spikezoo/pipeline/train_pipeline.py,sha256=BgHUsdv33B_OKauOVclNt7yIPb-_O-93ZHLHIjrwWaA,8459
 spikezoo/utils/__init__.py,sha256=bYLlusAXwLCoY4s6nhVgviax9ioRA9aea8qgRmj2HpI,152
 spikezoo/utils/data_utils.py,sha256=mk1xeyIb7o_E1J7Z6-gtPq-rpKiMTxAWSTcvvPvVku8,2033
@@ -166,8 +166,8 @@ spikezoo/utils/other_utils.py,sha256=uWNWaII9Jv7fkWNfkAD9wD-4ID-GAzbR-gGYT-1FF_c
 spikezoo/utils/scheduler_utils.py,sha256=5RBh-hl3-2y-IomxMs47T1p3JsbicZNYLza6q1uAKHo,828
 spikezoo/utils/spike_utils.py,sha256=u4Haa6Sp5xFqs61ztvq161oXTA_aZmNW3VYUZcayNW0,4296
 spikezoo/utils/vidar_loader.cpython-39-x86_64-linux-gnu.so,sha256=uXqu7ME---cZRRU5LUcLiNrjjtlOjxNwWHyTIQ10BGg,199088
-spikezoo-0.2.3.3.dist-info/LICENSE.txt,sha256=ukEi8E0PKq1dQGTXHUflg3rppLymwAhr7il9x-0nPgg,1062
-spikezoo-0.2.3.3.dist-info/METADATA,sha256=3aqIRpJr6TAfjldU8ZLmWy6uuCUla400W5tGpkH-X2M,11941
-spikezoo-0.2.3.3.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-spikezoo-0.2.3.3.dist-info/top_level.txt,sha256=xF2iuOstrACJh43NW4dsTwIdgKfXPXAb_Xzl3M1ricM,9
-spikezoo-0.2.3.3.dist-info/RECORD,,
+spikezoo-0.2.3.4.dist-info/LICENSE.txt,sha256=ukEi8E0PKq1dQGTXHUflg3rppLymwAhr7il9x-0nPgg,1062
+spikezoo-0.2.3.4.dist-info/METADATA,sha256=I2zSpBAJRP1VXpWb2X-QiUK-6YLX2EkVj00sEzgGPuY,11941
+spikezoo-0.2.3.4.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+spikezoo-0.2.3.4.dist-info/top_level.txt,sha256=xF2iuOstrACJh43NW4dsTwIdgKfXPXAb_Xzl3M1ricM,9
+spikezoo-0.2.3.4.dist-info/RECORD,,