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 {
return stringBuilder.toString();
}
public void include(String relativePath, String templateWithoutFileEnding, Writer writer, NetDefinitionMode netDefinitionMode){
......@@ -39,6 +40,12 @@ public class CNNArch2GluonTemplateController extends CNNArchTemplateController {
ftlContext.put(TEMPLATE_CONTROLLER_KEY, this);
ftlContext.put(ELEMENT_DATA_KEY, getCurrentElement());
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);
}
......
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