Skip to content
Snippets Groups Projects
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
No related branches found
No related tags found
1 merge request!23Added Unroll-related features and layers
Pipeline #178369 failed
......@@ -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})
......
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