Commit 0b7fb88f authored by lr119628's avatar lr119628
Browse files

[update] python code fix

parent 828967a3
Pipeline #524633 failed with stage
in 30 seconds
......@@ -15,6 +15,7 @@ from numpy import log, product,prod,sqrt
from mxnet.ndarray import zeros,zeros_like
sys.path.insert(1, '${tc.architecture.getAdaNetUtils()}')
from AdaNetConfig import AdaNetConfig
import CoreAdaNet
</#if>
<#if tc.architecture.customPyFilesPath??>
sys.path.insert(1, '${tc.architecture.customPyFilesPath}')
......@@ -570,6 +571,8 @@ class DataClass_${networkInstruction?index}:
</#if>
<#if block.isPresent()>
self.block = ${block.get().name}
if self.block is AdaNetConfig.DEFAULT_BLOCK.value:
self.block = CoreAdaNet.DefaultBuildingBlock
<#else>
self.block = None
</#if>
......
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