diff --git a/python/paddle/fluid/dygraph/static_runner.py b/python/paddle/fluid/dygraph/static_runner.py index 779bbdb73df7d75d7ebdf34cd7303101b4367e20..aff00b61e5c3e6aa2aca6b0ea6a4f3ce8ede8e72 100644 --- a/python/paddle/fluid/dygraph/static_runner.py +++ b/python/paddle/fluid/dygraph/static_runner.py @@ -198,14 +198,10 @@ class StaticModelRunner(layers.Layer): self._recheck_stop_gradients() def train(self): - # TODO: remove global train_mode setting - framework._dygraph_tracer().train_mode() self._is_test = False self._change_is_test_status(False) def eval(self): - # TODO: remove global train_mode setting - framework._dygraph_tracer().eval_mode() self._is_test = True self._change_is_test_status(True)