spikezoo 0.2.3.6__tar.gz → 0.2.3.7__tar.gz

{spikezoo-0.2.3.6 → spikezoo-0.2.3.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: spikezoo
-Version: 0.2.3.6
+Version: 0.2.3.7
 Summary: A deep learning toolbox for spike-to-image models.
 Home-page: https://github.com/chenkang455/Spike-Zoo
 Author: Kang Chen
@@ -41,7 +41,7 @@ Dynamic: summary
 
 <h5 align="center">
 
-[![GitHub repo stars](https://img.shields.io/github/stars/chenkang455/Spike-Zoo?style=flat&logo=github&logoColor=whitesmoke&label=Stars)](https://github.com/chenkang455/Spike-Zoo/stargazers) [![GitHub Issues](https://img.shields.io/github/issues/chenkang455/Spike-Zoo?style=flat&logo=github&logoColor=whitesmoke&label=Stars)](https://github.com/chenkang455/Spike-Zoo/issues) <a href="https://badge.fury.io/py/spikezoo"><img src="https://badge.fury.io/py/spikezoo.svg" alt="PyPI version"></a>  <a href='https://spike-zoo.readthedocs.io/zh-cn/latest/index.html'><img src='https://readthedocs.com/projects/plenoptix-nerfstudio/badge/?version=latest' alt='Documentation Status' /></a>[![License](https://img.shields.io/badge/License-MIT-yellow)](https://github.com/chenkang455/Spike-Zoo)
+[![GitHub repo stars](https://img.shields.io/github/stars/chenkang455/Spike-Zoo?style=flat&logo=github&logoColor=whitesmoke&label=Stars)](https://github.com/chenkang455/Spike-Zoo/stargazers) [![GitHub Issues](https://img.shields.io/github/issues/chenkang455/Spike-Zoo?style=flat&logo=github&logoColor=whitesmoke&label=Issues)](https://github.com/chenkang455/Spike-Zoo/issues) <a href="https://badge.fury.io/py/spikezoo"><img src="https://badge.fury.io/py/spikezoo.svg" alt="PyPI version"></a>  <a href='https://spike-zoo.readthedocs.io/zh-cn/latest/index.html'><img src='https://readthedocs.com/projects/plenoptix-nerfstudio/badge/?version=latest' alt='Documentation Status' /></a>[![License](https://img.shields.io/badge/License-MIT-yellow)](https://github.com/chenkang455/Spike-Zoo)
 <p>
 
 

spikezoo-0.2.3.7/spikezoo/pipeline/train_cfgs.py

@@ -0,0 +1,70 @@
+import torch.nn as nn
+import torch.optim as optimizer
+import torch.optim.lr_scheduler as lr_scheduler
+import functools
+from spikezoo.utils.optimizer_utils import OptimizerConfig, AdamOptimizerConfig
+from spikezoo.utils.scheduler_utils import SchedulerConfig, MultiStepSchedulerConfig, CosineAnnealingLRConfig
+from dataclasses import dataclass, field
+from spikezoo.pipeline.train_pipeline import TrainPipelineConfig
+from typing import Optional, Dict, List
+
+
+@dataclass
+class REDS_BASE_TrainConfig(TrainPipelineConfig):
+    """Training setting for methods on the REDS-BASE dataset."""
+
+    # parameters setting
+    epochs: int = 600
+    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", "lpips", "niqe", "brisque", "piqe"])
+
+    # dataloader setting
+    bs_train: int = 8
+    nw_train: int = 4
+    pin_memory: bool = False
+
+    # train setting - optimizer & scheduler & loss_dict
+    optimizer_cfg: OptimizerConfig = field(default_factory=lambda: AdamOptimizerConfig(lr=1e-4))
+    scheduler_cfg: Optional[SchedulerConfig] = field(
+        default_factory=lambda: MultiStepSchedulerConfig(milestones=[400], 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):
+#     """Training setting for BSF. https://github.com/ruizhao26/BSF"""
+
+#     optimizer_cfg: OptimizerConfig = AdamOptimizerConfig(lr=1e-4, weight_decay=0.0)
+#     scheduler_cfg: Optional[SchedulerConfig] = MultiStepSchedulerConfig(milestones=[10, 20, 30, 40, 50, 60, 70, 80, 90, 100])
+#     loss_weight_dict: Dict = field(default_factory=lambda: {"l1": 1})
+
+
+# @dataclass
+# class WGSETrainConfig(TrainPipelineConfig):
+#     """Training setting for WGSE. https://github.com/Leozhangjiyuan/WGSE-SpikeCamera"""
+
+#     optimizer_cfg: OptimizerConfig = AdamOptimizerConfig(lr=1e-4, betas=(0.9, 0.99), weight_decay=0)
+#     scheduler_cfg: Optional[SchedulerConfig] = MultiStepSchedulerConfig(milestones=[400, 600], gamma=0.2)
+#     loss_weight_dict: Dict = field(default_factory=lambda: {"l1": 1})
+
+
+# @dataclass
+# class STIRTrainConfig(TrainPipelineConfig):
+#     """Training setting for STIR. https://github.com/GitCVfb/STIR"""
+
+#     optimizer_cfg: OptimizerConfig = AdamOptimizerConfig(lr=1e-4, betas=(0.9, 0.999))
+#     scheduler_cfg: Optional[SchedulerConfig] = MultiStepSchedulerConfig(milestones=[20, 25, 30, 35, 40, 45, 50, 55, 60, 65, 70], gamma=0.7)
+#     loss_weight_dict: Dict = field(default_factory=lambda: {"l1": 1})
+
+
+# @dataclass
+# class Spk2ImgNetTrainConfig(TrainPipelineConfig):
+#     """Training setting for Spk2ImgNet. https://github.com/Vspacer/Spk2ImgNet"""
+
+#     optimizer_cfg: OptimizerConfig = AdamOptimizerConfig(lr=1e-4)
+#     scheduler_cfg: Optional[SchedulerConfig] = MultiStepSchedulerConfig(milestones=[20], gamma=0.1)
+#     loss_weight_dict: Dict = field(default_factory=lambda: {"l1": 1})

{spikezoo-0.2.3.6 → spikezoo-0.2.3.7}/spikezoo/pipeline/train_pipeline.py

@@ -52,7 +52,7 @@ class TrainPipelineConfig(PipelineConfig):
 
     # train setting - optimizer & scheduler & loss_dict
     "Optimizer config."
-    optimizer_cfg: OptimizerConfig = AdamOptimizerConfig(lr=1e-3)
+    optimizer_cfg: OptimizerConfig = field(default_factory=lambda: AdamOptimizerConfig(lr=1e-3))
     "Scheduler config."
     scheduler_cfg: Optional[SchedulerConfig] = None
     "Loss dict {loss_name,weight}."
@@ -82,7 +82,7 @@ class TrainPipeline(Pipeline):
         """Model and Data setup."""
         # model
         self.model: BaseModel = build_model_name(model_cfg) if isinstance(model_cfg, str) else build_model_cfg(model_cfg)
-        self.model.build_network(mode = "train",version="local")
+        self.model.build_network(mode="train", version="local")
         torch.set_grad_enabled(True)
         # data
         if isinstance(dataset_cfg, str):
@@ -94,7 +94,7 @@ class TrainPipeline(Pipeline):
         self.train_dataset.build_source("train")
         self.dataset.build_source("test")
         self.train_dataloader = build_dataloader(self.train_dataset, self.cfg)
-        self.dataloader = build_dataloader(self.dataset,self.cfg)
+        self.dataloader = build_dataloader(self.dataset, self.cfg)
         # device
         self.device = "cuda" if torch.cuda.is_available() else "cpu"
 

{spikezoo-0.2.3.6 → spikezoo-0.2.3.7}/spikezoo.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: spikezoo
-Version: 0.2.3.6
+Version: 0.2.3.7
 Summary: A deep learning toolbox for spike-to-image models.
 Home-page: https://github.com/chenkang455/Spike-Zoo
 Author: Kang Chen
@@ -41,7 +41,7 @@ Dynamic: summary
 
 <h5 align="center">
 
-[![GitHub repo stars](https://img.shields.io/github/stars/chenkang455/Spike-Zoo?style=flat&logo=github&logoColor=whitesmoke&label=Stars)](https://github.com/chenkang455/Spike-Zoo/stargazers) [![GitHub Issues](https://img.shields.io/github/issues/chenkang455/Spike-Zoo?style=flat&logo=github&logoColor=whitesmoke&label=Stars)](https://github.com/chenkang455/Spike-Zoo/issues) <a href="https://badge.fury.io/py/spikezoo"><img src="https://badge.fury.io/py/spikezoo.svg" alt="PyPI version"></a>  <a href='https://spike-zoo.readthedocs.io/zh-cn/latest/index.html'><img src='https://readthedocs.com/projects/plenoptix-nerfstudio/badge/?version=latest' alt='Documentation Status' /></a>[![License](https://img.shields.io/badge/License-MIT-yellow)](https://github.com/chenkang455/Spike-Zoo)
+[![GitHub repo stars](https://img.shields.io/github/stars/chenkang455/Spike-Zoo?style=flat&logo=github&logoColor=whitesmoke&label=Stars)](https://github.com/chenkang455/Spike-Zoo/stargazers) [![GitHub Issues](https://img.shields.io/github/issues/chenkang455/Spike-Zoo?style=flat&logo=github&logoColor=whitesmoke&label=Issues)](https://github.com/chenkang455/Spike-Zoo/issues) <a href="https://badge.fury.io/py/spikezoo"><img src="https://badge.fury.io/py/spikezoo.svg" alt="PyPI version"></a>  <a href='https://spike-zoo.readthedocs.io/zh-cn/latest/index.html'><img src='https://readthedocs.com/projects/plenoptix-nerfstudio/badge/?version=latest' alt='Documentation Status' /></a>[![License](https://img.shields.io/badge/License-MIT-yellow)](https://github.com/chenkang455/Spike-Zoo)
 <p>
 
 

spikezoo-0.2.3.6/spikezoo/pipeline/train_cfgs.py

@@ -1,69 +0,0 @@
-import torch.nn as nn
-import torch.optim as optimizer
-import torch.optim.lr_scheduler as lr_scheduler
-import functools
-from spikezoo.utils.optimizer_utils import OptimizerConfig, AdamOptimizerConfig
-from spikezoo.utils.scheduler_utils import SchedulerConfig, MultiStepSchedulerConfig, CosineAnnealingLRConfig
-from dataclasses import dataclass, field
-from spikezoo.pipeline.train_pipeline import TrainPipelineConfig
-from typing import Optional, Dict, List
-
-
-@dataclass
-class REDS_BASE_TrainConfig(TrainPipelineConfig):
-    """Training setting for methods on the REDS-BASE dataset."""
-
-    # parameters setting
-    epochs: int = 600
-    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","lpips","niqe","brisque","piqe"])
-
-    # dataloader setting
-    bs_train: int = 8
-    nw_train: 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], 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):
-    """Training setting for BSF. https://github.com/ruizhao26/BSF"""
-
-    optimizer_cfg: OptimizerConfig = AdamOptimizerConfig(lr=1e-4, weight_decay=0.0)
-    scheduler_cfg: Optional[SchedulerConfig] = MultiStepSchedulerConfig(milestones=[10, 20, 30, 40, 50, 60, 70, 80, 90, 100])
-    loss_weight_dict: Dict = field(default_factory=lambda: {"l1": 1})
-
-
-@dataclass
-class WGSETrainConfig(TrainPipelineConfig):
-    """Training setting for WGSE. https://github.com/Leozhangjiyuan/WGSE-SpikeCamera"""
-
-    optimizer_cfg: OptimizerConfig = AdamOptimizerConfig(lr=1e-4, betas=(0.9, 0.99), weight_decay=0)
-    scheduler_cfg: Optional[SchedulerConfig] = MultiStepSchedulerConfig(milestones=[400, 600], gamma=0.2)
-    loss_weight_dict: Dict = field(default_factory=lambda: {"l1": 1})
-
-
-@dataclass
-class STIRTrainConfig(TrainPipelineConfig):
-    """Training setting for STIR. https://github.com/GitCVfb/STIR"""
-
-    optimizer_cfg: OptimizerConfig = AdamOptimizerConfig(lr=1e-4, betas=(0.9, 0.999))
-    scheduler_cfg: Optional[SchedulerConfig] = MultiStepSchedulerConfig(milestones=[20, 25, 30, 35, 40, 45, 50, 55, 60, 65, 70], gamma=0.7)
-    loss_weight_dict: Dict = field(default_factory=lambda: {"l1": 1})
-
-
-@dataclass
-class Spk2ImgNetTrainConfig(TrainPipelineConfig):
-    """Training setting for Spk2ImgNet. https://github.com/Vspacer/Spk2ImgNet"""
-
-    optimizer_cfg: OptimizerConfig = AdamOptimizerConfig(lr=1e-4)
-    scheduler_cfg: Optional[SchedulerConfig] = MultiStepSchedulerConfig(milestones=[20], gamma=0.1)
-    loss_weight_dict: Dict = field(default_factory=lambda: {"l1": 1})