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

doc: add load to pytorch estimator quickstart. (#6920)

Browse source 
* doc: add load to pytorch estimator quickstart.

* fix: fix style.

* fix typo

* fix wording.
This commit is contained in:
Cengguang Zhang 2022-12-13 14:31:15 +08:00 committed by GitHub
parent 664dfbe7ef
commit de88f1e6b3
1 changed files with 12 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
docs/readthedocs/source/doc/Orca/QuickStart/orca-pytorch-quickstart.md
View file

@ -127,15 +127,22 @@ for r in result:
print(r, ":", result[r]) print(r, ":", result[r])
``` ```
### Step 5: Save the Model ### Step 5: Save and Load the Model
Save the Estimator states (including model and optimizer) to the provided model path.
Save the Estimator states (including the model and optimizer) to the provided model path.
```python ```python
est.save("mnist_model") est.save("mnist_model")
```
# stop orca context when program finishes Save the entire model to the provided model path.
stop_orca_context() ```python
est.save("mnist_model", entire=True)
```
Load the Estimator states (including the model and optimizer) or the entire model from the provided model path.
```python
est.load("mnist_model")
``` ```
**Note:** You should call `stop_orca_context()` when your application finishes. **Note:** You should call `stop_orca_context()` when your application finishes.