From ab0f4084c3a8561e563f3a652f0a10d003b1d5ec Mon Sep 17 00:00:00 2001 From: David Rotermund <54365609+davrot@users.noreply.github.com> Date: Sun, 23 Jul 2023 01:45:59 +0200 Subject: [PATCH] Add files via upload --- cnn_training.py | 3 +++ config.json | 5 +++-- 2 files changed, 6 insertions(+), 2 deletions(-) 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)