From 862c530685fa6d7a939196c219f631185f260ef5 Mon Sep 17 00:00:00 2001 From: dding3 Date: Tue, 21 Dec 2021 23:38:18 -0800 Subject: [PATCH] add get model in nnframe doc (#3776) * add get model in nnframe doc --- .../source/doc/DLlib/Overview/nnframes.md | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/docs/readthedocs/source/doc/DLlib/Overview/nnframes.md b/docs/readthedocs/source/doc/DLlib/Overview/nnframes.md index 23cddba0..dcfcec23 100644 --- a/docs/readthedocs/source/doc/DLlib/Overview/nnframes.md +++ b/docs/readthedocs/source/doc/DLlib/Overview/nnframes.md @@ -203,6 +203,18 @@ nn_model = NNModel(bigDLModel) Meanwhile, for advanced use cases (e.g. model with multiple input tensor), `NNModel` supports: `setSamplePreprocessing(value: Preprocessing[Any, Sample[T]])`to directly compose Sample according to user-specified Preprocessing. +We can get model from `NNModel` by: + +**Scala:** +```scala +val model = nnModel.getModel() +``` + +**Python:** +```python +model = nn_model.model +``` + --- ### 2.3 NNClassifier