Commit 1c37d990 authored by Evgeny Kusmenko's avatar Evgeny Kusmenko
Browse files

Merge branch 'MyElu-fix' into 'master'

fix myELU

See merge request !45
parents 3052564d b1e9eeee
Pipeline #531613 passed with stages
in 2 minutes and 6 seconds
...@@ -31,6 +31,7 @@ public class CNNArch2GluonTemplateController extends CNNArchTemplateController { ...@@ -31,6 +31,7 @@ public class CNNArch2GluonTemplateController extends CNNArchTemplateController {
return stringBuilder.toString(); return stringBuilder.toString();
} }
public void include(String relativePath, String templateWithoutFileEnding, Writer writer, NetDefinitionMode netDefinitionMode){ public void include(String relativePath, String templateWithoutFileEnding, Writer writer, NetDefinitionMode netDefinitionMode){
...@@ -39,6 +40,12 @@ public class CNNArch2GluonTemplateController extends CNNArchTemplateController { ...@@ -39,6 +40,12 @@ public class CNNArch2GluonTemplateController extends CNNArchTemplateController {
ftlContext.put(TEMPLATE_CONTROLLER_KEY, this); ftlContext.put(TEMPLATE_CONTROLLER_KEY, this);
ftlContext.put(ELEMENT_DATA_KEY, getCurrentElement()); ftlContext.put(ELEMENT_DATA_KEY, getCurrentElement());
ftlContext.put(NET_DEFINITION_MODE_KEY, netDefinitionMode.toString()); ftlContext.put(NET_DEFINITION_MODE_KEY, netDefinitionMode.toString());
if (this.getDataElement().getElement() instanceof LayerSymbol){
if(((LayerSymbol) (this.getDataElement().getElement())).getDeclaration() instanceof CustomLayerDeclaration){
templatePath = relativePath + "CustomLayer" + FTL_FILE_ENDING;
}
}
getTemplateConfiguration().processTemplate(ftlContext, templatePath, writer); getTemplateConfiguration().processTemplate(ftlContext, templatePath, writer);
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment