ayo/ipex-llm
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

add get model in nnframe doc (#3776)

Browse source 
* add get model in nnframe doc
This commit is contained in:
dding3 2021-12-21 23:38:18 -08:00 committed by GitHub
parent f27f2e9d3a
commit 862c530685
1 changed files with 12 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
docs/readthedocs/source/doc/DLlib/Overview/nnframes.md
View file

@ -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. 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 ### 2.3 NNClassifier