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
......@@ -33,12 +33,19 @@ public class CNNArch2GluonTemplateController extends CNNArchTemplateController {
public void include(String relativePath, String templateWithoutFileEnding, Writer writer, NetDefinitionMode netDefinitionMode){
String templatePath = relativePath + templateWithoutFileEnding + FTL_FILE_ENDING;
Map<String, Object> ftlContext = new HashMap<>();
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