Example #1
0
 def optimize(self):
     """
     Do an optimization.
     """
     jmodel = callJavaFunc(self.value.optimize)
     from bigdl.dllib.nn.layer import Layer
     return Layer.of(jmodel)
Example #2
0
 def flattened_layers(self, include_container=False):
     jlayers = callZooFunc(self.bigdl_type, "getFlattenSubModules", self,
                           include_container)
     layers = [Layer.of(jlayer) for jlayer in jlayers]
     return layers
Example #3
0
 def layers(self):
     jlayers = callZooFunc(self.bigdl_type, "getSubModules", self)
     layers = [Layer.of(jlayer) for jlayer in jlayers]
     return layers