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

[update] update

parent 30fdf01a
......@@ -16,7 +16,7 @@ from custom_layers import *
<#list tc.architecture.networkInstructions as networkInstruction>
<#if tc.containsAdaNet()>
from CNNNet_${tc.fullArchitectureName} import Net_${networkInstruction?index},DataClass
from CNNNet_${tc.fullArchitectureName} import Net_${networkInstruction?index},DataClass_${networkInstruction?index}
<#else>
from CNNNet_${tc.fullArchitectureName} import Net_${networkInstruction?index}
</#if>
......@@ -178,8 +178,8 @@ class ${tc.fileNameWithoutEnding}:
def construct(self, context, data_mean=None, data_std=None):
<#list tc.architecture.networkInstructions as networkInstruction>
<#if tc.containsAdaNet()>
self.networks[${networkInstruction?index}] = Net_${networkInstruction?index}(prefix="")
self.dataClass = DataCLass()
self.networks[${networkInstruction?index}] = Net_${networkInstruction?index}(prefix="",operations=None)
self.dataClass = DataClass_${networkInstruction?index}()
<#else>
self.networks[${networkInstruction?index}] = Net_${networkInstruction?index}(data_mean=data_mean, data_std=data_std, mx_context=context, prefix="")
</#if>
......
......@@ -570,11 +570,10 @@ class Net_${networkInstruction?index}(gluon.HybridBlock):
mean_complexity[i] = self.candidate_complexities[name]
return mean_complexity
class DataClass():
class DataClass_${networkInstruction?index}:
"""
the whole model with its operations
"""
def __init__(self, **kwargs):
self.op_names = [] # list that holds the name of the added operations
self.candidate_complexities = {}
......
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