update NeuralNet

MartinAchondo · Jul 11, 2024 · 422ad34 · 422ad34
1 parent 61eba3e
commit 422ad34
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/xppbe/NN/NeuralNet.py b/xppbe/NN/NeuralNet.py
@@ -126,7 +126,7 @@ def __init__(self,
             self.create_ResNet()
 
         # Output layer
-        self.out = tf.keras.layers.Dense(output_dim,
+        self.out = self.Dense_Layer(output_dim,
                                activation=None,
                                name=f'output_layer')