spikezoo 0.2.3.3__tar.gz → 0.2.3.4__tar.gz

{spikezoo-0.2.3.3 → spikezoo-0.2.3.4}/PKG-INFO

@@ -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 → spikezoo-0.2.3.4}/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-0.2.3.3 → spikezoo-0.2.3.4}/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 → spikezoo-0.2.3.4}/spikezoo.egg-info/PKG-INFO

@@ -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