|
|
@ -49,10 +49,10 @@ class Layer(core.Layer):
|
|
|
|
self._helper = LayerObjectHelper(self._full_name)
|
|
|
|
self._helper = LayerObjectHelper(self._full_name)
|
|
|
|
|
|
|
|
|
|
|
|
def train(self):
|
|
|
|
def train(self):
|
|
|
|
framework._dygraph_tracer()._train_mode()
|
|
|
|
framework._dygraph_tracer().train_mode()
|
|
|
|
|
|
|
|
|
|
|
|
def eval(self):
|
|
|
|
def eval(self):
|
|
|
|
framework._dygraph_tracer()._eval_mode()
|
|
|
|
framework._dygraph_tracer().eval_mode()
|
|
|
|
|
|
|
|
|
|
|
|
def full_name(self):
|
|
|
|
def full_name(self):
|
|
|
|
"""Full name for this layers.
|
|
|
|
"""Full name for this layers.
|
|
|
@ -261,10 +261,10 @@ class PyLayer(core.PyLayer):
|
|
|
|
super(PyLayer, self).__init__()
|
|
|
|
super(PyLayer, self).__init__()
|
|
|
|
|
|
|
|
|
|
|
|
def train(self):
|
|
|
|
def train(self):
|
|
|
|
framework._dygraph_tracer()._train_mode()
|
|
|
|
framework._dygraph_tracer().train_mode()
|
|
|
|
|
|
|
|
|
|
|
|
def eval(self):
|
|
|
|
def eval(self):
|
|
|
|
framework._dygraph_tracer()._eval_mode()
|
|
|
|
framework._dygraph_tracer().eval_mode()
|
|
|
|
|
|
|
|
|
|
|
|
@classmethod
|
|
|
|
@classmethod
|
|
|
|
def _do_forward(cls, inputs):
|
|
|
|
def _do_forward(cls, inputs):
|
|
|
|