Allow sigmoid

4ed3b332 · Nicola Gatto · 9251bd58 · 4ed3b332
Commit 4ed3b332 authored Jun 10, 2019 by Nicola Gatto
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/main/java/de/monticore/lang/monticar/cnnarch/gluongenerator/CNNArch2GluonTemplateController.java ...narch/gluongenerator/CNNArch2GluonTemplateController.java +1 -1

No files found.
--- a/src/main/java/de/monticore/lang/monticar/cnnarch/gluongenerator/CNNArch2GluonTemplateController.java
+++ b/src/main/java/de/monticore/lang/monticar/cnnarch/gluongenerator/CNNArch2GluonTemplateController.java
@@ -71,7 +71,7 @@ public class CNNArch2GluonTemplateController extends CNNArchTemplateController {

        if (layer.isAtomic()){
            ArchitectureElementSymbol nextElement = layer.getOutputElement().get();
-            if (!isSoftmaxOutput(nextElement) && !isLogisticRegressionOutput(nextElement)){
+            if (!isSoftmaxOutput(nextElement)){
                String templateName = layer.getDeclaration().getName();
                include(TEMPLATE_ELEMENTS_DIR_PATH, templateName, writer, netDefinitionMode);
            }