Commit dd120e69 authored by lr119628's avatar lr119628
Browse files

[update] check if networkinstruction contains adaNet

parent 20215ef4
......@@ -691,7 +691,7 @@ class ${tc.fileNameWithoutEnding}:
loss_function.hybridize()
<#if tc.containsAdaNet()>
<#list tc.architecture.networkInstructions as networkInstruction>
<#if networkInstruction.containsAdaNet()>
assert self._networks[${networkInstruction?index}].AdaNet, "passed model is not an AdaNet model"
self._networks[${networkInstruction?index}] = fit(
loss=loss_function,
......@@ -709,6 +709,7 @@ class ${tc.fileNameWithoutEnding}:
)
logging.info(self._networks[0])
#put here the AdaNet logic
</#if>
</#list>
</#if>
<#list tc.architecture.networkInstructions as networkInstruction>
......
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