diff --git a/mindspore/train/model.py b/mindspore/train/model.py index 896a60b8e98..ff32720e404 100755 --- a/mindspore/train/model.py +++ b/mindspore/train/model.py @@ -842,5 +842,20 @@ class Model: if param.cache_enable: Tensor(param).flush_from_cache() + @property + def train_network(self): + """Get the model's train_network.""" + return self._train_network + + @property + def predict_network(self): + """Get the model's predict_network.""" + return self._predict_network + + @property + def eval_network(self): + """Get the model's eval_network.""" + return self._eval_network + __all__ = ["Model"]