Commit 91473f2b authored by Carlos Alfredo Yeverino Rodriguez's avatar Carlos Alfredo Yeverino Rodriguez
Browse files

Corrected Sigmoid layer template. Minor esthetic change in Relu layer template

parent 5608cd10
<#assign input = element.inputs[0]> <#assign input = element.inputs[0]>
${} = brew.relu(model, ${input}, ${input}) ${} = brew.relu(model, ${input}, ${input})
${} = mx.symbol.Activation(data=${element.inputs[0]}, <#assign input = element.inputs[0]>
act_type='sigmoid', ${} =${input}, '${}')
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