Commit 400b129f authored by Carlos Alfredo Yeverino Rodriguez's avatar Carlos Alfredo Yeverino Rodriguez
Browse files

Removed info comments. Minor esthetic change in outputshape.

parent 8b247cb3
......@@ -22,7 +22,4 @@
${element.name} = brew.conv(model, ${input}, '${element.name}', dim_in=${element.element.inputTypes[0].channels?c}, dim_out=${element.channels?c}, ${kernelParameter}, ${strideParameter})
</#if>
<#-- TODO: check how to adapt CNNArchLang argument no_bias=${element.noBias?string("True","False")} -->
<#include "OutputShape.ftl">
# Yeverino input layer name: ${element.inputs[0]}
# Yeverino input/previous layer dim_out: ${element.element.inputTypes[0].channels?c}
# Yeverino current layer dim_out: ${element.element.outputTypes[0].channels?c}
<#include "OutputShape.ftl">
\ No newline at end of file
......@@ -13,6 +13,4 @@
${element.name} = brew.fc(model, ${input}, '${element.name}', dim_in=${inputChannels} * ${inputHeight} * ${inputWidth}, dim_out=${element.units?c})
</#if>
<#-- TODO: check how to adapt CNNArchLang argument no_bias=${element.noBias?string("True","False")} -->
# Yeverino input layer type: ${inputLayerType}
# Yeverino current layer type: ${element.element}
<#include "OutputShape.ftl">
\ No newline at end of file
# ${element.name}, output shape: {<#list element.element.outputTypes as type>[${tc.join(type.dimensions, ",")}]</#list>
# ${element.name}, output shape: {<#list element.element.outputTypes as type>[${tc.join(type.dimensions, ",")}]</#list>}
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