Commit 97212579 authored by Christian Fuß's avatar Christian Fuß
Browse files

fixed small bug for accessing the same layer again in nets using unrolls

parent c2f2d411
Pipeline #178369 failed with stages
in 2 minutes and 18 seconds
......@@ -3,14 +3,14 @@
<#assign use_bias = element.noBias?string("False","True")>
<#assign flatten = element.flatten?string("True","False")>
<#if mode == "ARCHITECTURE_DEFINITION">
<#if (element.partOfUnroll && false)>
<#if element.partOfUnroll>
${element.name} = Net_1().${element.name}(${input})
<#else>
self.${element.name} = gluon.nn.Dense(units=${units}, use_bias=${use_bias}, flatten=${flatten})
</#if>
<#include "OutputShape.ftl">
<#elseif mode == "FORWARD_FUNCTION">
<#if (element.partOfUnroll && false)>
<#if element.partOfUnroll>
${element.name} = Net_1().${element.name}(${input})
<#else>
${element.name} = self.${element.name}(${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