forked from mindspore-Ecosystem/mindspore
add model attr
This commit is contained in:
parent
0fdf5c54e3
commit
441aca02f7
|
@ -842,5 +842,20 @@ class Model:
|
||||||
if param.cache_enable:
|
if param.cache_enable:
|
||||||
Tensor(param).flush_from_cache()
|
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"]
|
__all__ = ["Model"]
|
||||||
|
|
Loading…
Reference in New Issue