Skip to content
Snippets Groups Projects
Commit 4ed3b332 authored by Nicola Gatto's avatar Nicola Gatto
Browse files

Allow sigmoid

parent 9251bd58
No related branches found
No related tags found
1 merge request!17Shared code, updated for CNNArchLang, etc.
Pipeline #148817 failed
......@@ -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);
}
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment