Browse Source

!13439 add model attr train/predict network

From: @changzherui
Reviewed-by: @kingxian,@zhoufeng54
Signed-off-by: @kingxian
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 5 years ago
parent
commit
ee7fcccec6
1 changed files with 15 additions and 0 deletions
  1. +15
    -0
      mindspore/train/model.py

+ 15
- 0
mindspore/train/model.py View File

@@ -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"]

Loading…
Cancel
Save