diff --git a/cnn_training.py b/cnn_training.py index 636b7b3..f69c795 100644 --- a/cnn_training.py +++ b/cnn_training.py @@ -75,6 +75,7 @@ def main( pooling_type=str(config["pooling_type"]), conv_0_enable_softmax=bool(config["conv_0_enable_softmax"]), conv_0_power_softmax=float(config["conv_0_power_softmax"]), + conv_0_meanmode_softmax=bool(config["conv_0_meanmode_softmax"]), scale_data=int(config["scale_data"]), use_scheduler=bool(config["use_scheduler"]), use_adam=bool(config["use_adam"]), @@ -118,6 +119,7 @@ def run_network( pooling_type: str, conv_0_enable_softmax: bool, conv_0_power_softmax: float, + conv_0_meanmode_softmax: bool, scale_data: float, use_scheduler: bool, use_adam: bool, @@ -246,6 +248,7 @@ def run_network( conv_0_enable_softmax=conv_0_enable_softmax, l_relu_negative_slope=leak_relu_negative_slope, conv_0_power_softmax=conv_0_power_softmax, + conv_0_meanmode_softmax=conv_0_meanmode_softmax, ).to(device) logger.info(model) diff --git a/config.json b/config.json index 76be00e..f47f0b4 100644 --- a/config.json +++ b/config.json @@ -27,11 +27,12 @@ "scheduler_patience": 100, // (10) "scheduler_threshold": 1e-5, // (1e-4) "minimum_learning_rate": 1e-10, - "learning_rate": 1e-3, + "learning_rate": 1e-4, // <- LR Scheduler "pooling_type": "max", // (max), average, none "conv_0_enable_softmax": true, // true, (false) - "conv_0_power_softmax": 2.0, // 0.0 = Exp + "conv_0_power_softmax": 0.0, // (0.0) = Exp + "conv_0_meanmode_softmax": false, // true, false "use_adam": true, // (true) => adam, false => SGD "condition": "Coignless", "scale_data": 255.0, // (255.0)