README.md

@@ -139,18 +139,6 @@ model.fit(train_dataset=trainset, test_dataset=testset, val_dataset=None)
 ```
 
 
-## Naming conventions
-
-Some naming conventions used in the code differ from the ones used in the paper:
-- CSNN: feature extraction layer
-- CSNN-1: CSNN-16 (16 feature maps)
-- CSNN-2: CSNN-64 (64 feature maps)
-- CSNN-3: CSNN-128 (128 feature maps)
-- Readout: classification layer
-- SSTDP-OG: SSTDPv1
-- SSTDP: SSTDPv2
-- PCN-SSTDP: SSTDPv2+PCN
-
 ## Known bugs
 
 - Depending on your machine, results on CIFAR-10 can slightly differ (not a significative difference though). It comes from the download/conversion process in `datasets/get_cifar10.py`.

scripts/config/CIFAR10/analysis/config/sstdp-og.json → scripts/config/CIFAR10/analysis/config/s2stdp.json

@@ -19,7 +19,7 @@
     "loss_type": "mean",
     "t_gap": [ 0.01, 0.001, 0.005, 0.01, 0.005, 0.01, 0.001, 0.01, 0.001, 0.005 ],
     "adjust_t_gap": true,
-    "loss_bounds": true,
+    "loss_bounds": false,
     "class_wta": true,
     "early_stopping": true,
     "early_stop_patience": 10,

scripts/config/CIFAR10/analysis/config/sstdp.json

@@ -19,7 +19,7 @@
     "loss_type": "mean",
     "t_gap": [ 0.01, 0.001, 0.005, 0.01, 0.005, 0.01, 0.001, 0.01, 0.001, 0.005 ],
     "adjust_t_gap": true,
-    "loss_bounds": false,
+    "loss_bounds": true,
     "class_wta": true,
     "early_stopping": true,
     "early_stop_patience": 10,

scripts/config/CIFAR10/analysis/lr/sstdp-og.json → scripts/config/CIFAR10/analysis/lr/s2stdp.json

@@ -2,7 +2,7 @@
     "full_logs": true,
 
     "n_neurons": 10,
-    "firing_threshold": 200,
+    "firing_threshold": 500,
     "leak_tau": null,
     "decision_making": "single",
 
@@ -12,14 +12,14 @@
     "w_init_mean": 0.5,
     "w_init_std": 0.01,
     "normalize_weights": true,
-    "w_mean_norm": 0.2,
+    "w_mean_norm": 0.3,
 
     "n_epochs": 100,
     "training_method": "sstdp",
     "loss_type": "mean",
-    "t_gap": 0.0005,
+    "t_gap": 0.005,
     "adjust_t_gap": true,
-    "loss_bounds": true,
+    "loss_bounds": false,
     "class_wta": true,
     "early_stopping": true,
     "early_stop_patience": 10,

scripts/config/CIFAR10/analysis/lr/sstdp.json

@@ -2,7 +2,7 @@
     "full_logs": true,
 
     "n_neurons": 10,
-    "firing_threshold": 500,
+    "firing_threshold": 200,
     "leak_tau": null,
     "decision_making": "single",
 
@@ -12,14 +12,14 @@
     "w_init_mean": 0.5,
     "w_init_std": 0.01,
     "normalize_weights": true,
-    "w_mean_norm": 0.3,
+    "w_mean_norm": 0.2,
 
     "n_epochs": 100,
     "training_method": "sstdp",
     "loss_type": "mean",
-    "t_gap": 0.005,
+    "t_gap": 0.0005,
     "adjust_t_gap": true,
-    "loss_bounds": false,
+    "loss_bounds": true,
     "class_wta": true,
     "early_stopping": true,
     "early_stop_patience": 10,

scripts/config/CIFAR10/analysis/t_gap/sstdp-og.json → scripts/config/CIFAR10/analysis/t_gap/s2stdp.json

@@ -2,7 +2,7 @@
     "full_logs": true,
 
     "n_neurons": 10,
-    "firing_threshold": 200,
+    "firing_threshold": 500,
     "leak_tau": null,
     "decision_making": "single",
 
@@ -12,14 +12,14 @@
     "w_init_mean": 0.5,
     "w_init_std": 0.01,
     "normalize_weights": true,
-    "w_mean_norm": 0.2,
+    "w_mean_norm": 0.3,
 
     "n_epochs": 100,
     "training_method": "sstdp",
     "loss_type": "mean",
     "t_gap": [ 0.2, 0.1, 0.075, 0.05, 0.025, 0.01, 0.0075, 0.005, 0.0025, 0.001, 0.00075, 0.0005, 0.00025, 0.0001 ],
     "adjust_t_gap": true,
-    "loss_bounds": true,
+    "loss_bounds": false,
     "class_wta": true,
     "early_stopping": true,
     "early_stop_patience": 10,
@@ -27,7 +27,7 @@
     "stdp_type": "multiplicative",
     "stdp_soft_bound": false,
     "stdp_ignore_silent": false,
-    "stdp_ap": 0.075,
+    "stdp_ap": 0.001,
     "stdp_am": -0.075,
     "stdp_anti_ap": null,
     "stdp_anti_am": null,

scripts/config/CIFAR10/analysis/t_gap/sstdp.json

@@ -2,7 +2,7 @@
     "full_logs": true,
 
     "n_neurons": 10,
-    "firing_threshold": 500,
+    "firing_threshold": 200,
     "leak_tau": null,
     "decision_making": "single",
 
@@ -12,14 +12,14 @@
     "w_init_mean": 0.5,
     "w_init_std": 0.01,
     "normalize_weights": true,
-    "w_mean_norm": 0.3,
+    "w_mean_norm": 0.2,
 
     "n_epochs": 100,
     "training_method": "sstdp",
     "loss_type": "mean",
     "t_gap": [ 0.2, 0.1, 0.075, 0.05, 0.025, 0.01, 0.0075, 0.005, 0.0025, 0.001, 0.00075, 0.0005, 0.00025, 0.0001 ],
     "adjust_t_gap": true,
-    "loss_bounds": false,
+    "loss_bounds": true,
     "class_wta": true,
     "early_stopping": true,
     "early_stop_patience": 10,
@@ -27,7 +27,7 @@
     "stdp_type": "multiplicative",
     "stdp_soft_bound": false,
     "stdp_ignore_silent": false,
-    "stdp_ap": 0.001,
+    "stdp_ap": 0.075,
     "stdp_am": -0.075,
     "stdp_anti_ap": null,
     "stdp_anti_am": null,