From 441aca02f7b3b9e1a7ee165eaafc4e91364e1110 Mon Sep 17 00:00:00 2001
From: changzherui <changzherui1@huawei.com>
Date: Tue, 16 Mar 2021 23:01:03 +0800
Subject: [PATCH] add model attr

---
 mindspore/train/model.py | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/mindspore/train/model.py b/mindspore/train/model.py
index 896a60b8e9..ff32720e40 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"]