if self.ops is None:
self.ops = util.get_ops(self.device)
self._root = kwargs.get("root", self)
self._parent = kwargs.get("parent", None)
if self._parent is not None:
self.name = self._parent.name + "-" + self.name
self.params = self.Params(self.ops)
After Change
self._layers = []
self._operators = {}
for name in self.desc.dimensions:
if not hasattr(self, name):
setattr(self, name, None)
self.desc.dimensions.set_from_init(self.args)
def begin_training(self, train_X, train_Y):
set_dimensions_given_data(self, train_X, train_Y)