From 39b341f262652038c1f81786d58b7fa7925d6470 Mon Sep 17 00:00:00 2001
From: David Rotermund <54365609+davrot@users.noreply.github.com>
Date: Sun, 23 Jul 2023 01:59:13 +0200
Subject: [PATCH] Add files via upload

---
 functions/make_cnn.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/functions/make_cnn.py b/functions/make_cnn.py
index f752e31..49d9edf 100644
--- a/functions/make_cnn.py
+++ b/functions/make_cnn.py
@@ -17,6 +17,7 @@ def make_cnn(
     conv_0_enable_softmax: bool,
     conv_0_power_softmax: float,
     conv_0_meanmode_softmax: bool,
+    conv_0_no_input_mode_softmax: bool,
     l_relu_negative_slope: float,
 ) -> torch.nn.Sequential:
     assert len(conv_out_channels_list) >= 1
@@ -63,7 +64,10 @@ def make_cnn(
     if conv_0_enable_softmax:
         cnn.append(
             SoftmaxPower(
-                dim=1, power=conv_0_power_softmax, mean_mode=conv_0_meanmode_softmax
+                dim=1,
+                power=conv_0_power_softmax,
+                mean_mode=conv_0_meanmode_softmax,
+                no_input_mode=conv_0_no_input_mode_softmax,
             )
         )