def optimize(self):
"""
Do an optimization.
"""
jmodel = callJavaFunc(self.value.optimize)
from bigdl.dllib.nn.layer import Layer
return Layer.of(jmodel)
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
def layers(self):
jlayers = callZooFunc(self.bigdl_type, "getSubModules", self)
layers = [Layer.of(jlayer) for jlayer in jlayers]
return layers