Commit 7c16c284 authored by lr119628's avatar lr119628
Browse files

[update] update

parent 7add7fcd
......@@ -29,6 +29,7 @@ class ${tc.fileNameWithoutEnding}:
def __init__(self):
self.weight_initializer = mx.init.Normal()
self.networks = {}
self.dataClass = {}
<#if (tc.weightsPath)??>
self._weights_dir_ = "${tc.weightsPath}/"
<#else>
......@@ -179,7 +180,7 @@ class ${tc.fileNameWithoutEnding}:
<#list tc.architecture.networkInstructions as networkInstruction>
<#if tc.containsAdaNet()>
self.networks[${networkInstruction?index}] = Net_${networkInstruction?index}(prefix="",operations=None)
self.dataClass = DataClass_${networkInstruction?index}()
self.dataClass[${networkInstruction?index}] = DataClass_${networkInstruction?index}()
<#else>
self.networks[${networkInstruction?index}] = Net_${networkInstruction?index}(data_mean=data_mean, data_std=data_std, mx_context=context, prefix="")
</#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