Skip to content
Snippets Groups Projects
Commit dec0229d authored by Christian Fuß's avatar Christian Fuß
Browse files

fixed indent in Embedding- and missing comma in LSTM template

parent 01c16d6c
No related branches found
No related tags found
1 merge request!23Added Unroll-related features and layers
Pipeline #181635 failed
<#assign input = element.inputs[0]>
<#if mode == "ARCHITECTURE_DEFINITION">
self.${element.name} = gluon.nn.Embedding(input_dim=${element.inputDim?c}, output_dim=${element.outputDim?c})
<#include "OutputShape.ftl">
self.${element.name} = gluon.nn.Embedding(input_dim=${element.inputDim?c}, output_dim=${element.outputDim?c})
<#include "OutputShape.ftl">
<#elseif mode == "FORWARD_FUNCTION">
${element.name} = self.${element.name}(${input})
</#if>
\ No newline at end of file
......@@ -3,7 +3,7 @@
<#if mode == "ARCHITECTURE_DEFINITION">
self.${element.name} = gluon.rnn.LSTM(hidden_size=${element.units?c},
num_layers=${element.layers?c},
bidirectional=${element.bidirectional?string("True", "False")}
bidirectional=${element.bidirectional?string("True", "False")},
layout='NTC')
<#include "OutputShape.ftl">
<#elseif mode == "FORWARD_FUNCTION">
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment