[update] update

src/main/resources/templates/gluon/CNNCreator.ftl

@@ -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>

src/main/resources/templates/gluon/CNNNet.ftl

@@ -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 = {}