Commit cdb5225f authored by lr119628's avatar lr119628
Browse files

[update] model has now correct output shape

parent 5f3d197d
......@@ -37,7 +37,17 @@ class CandidateHull(gluon.HybridBlock):
stack: int,
name: str,
output=None,
<#if outBlock.isArtificial()>
output_shape =<#list outBlock.outputTypes as type>(${tc.join(type.dimensions, ",")})</#list>,
<#else>
output_shape =None,
</#if>
inBlock=None,
<#if inBlock.isArtificial()>
input_shape =<#list inBlock.outputTypes as type>(${tc.join(type.dimensions, ",")})</#list>,
<#else>
input_shape =None,
</#if>
block_args=None,
**kwargs):
......
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