From 57d45e5deb3c65bb3c7e3db46b2b2f91ae9e35d0 Mon Sep 17 00:00:00 2001
From: Cengguang Zhang <potterguang101@gmail.com>
Date: Wed, 24 Aug 2022 13:48:32 +0800
Subject: [PATCH] Orca: add save and load model doc to PyTorch Estimator
 quick-start. (#5504)

* feat: add save and load model doc to pytorch estimator quickstart.

* fix: fix typo.

* fix: fix typo

* fix: fix typo.

* feat: add doc to ray backend quickstart

* fix: fix typo.

* feat: add doc to ray backend quickstart
---
 .../orca-pytorch-distributed-quickstart.md         | 14 ++++++++++++++
 .../doc/Orca/QuickStart/orca-pytorch-quickstart.md | 14 ++++++++++++++
 .../Orca/QuickStart/orca-tf2keras-quickstart.md    |  8 ++++----
 3 files changed, 32 insertions(+), 4 deletions(-)

diff --git a/docs/readthedocs/source/doc/Orca/QuickStart/orca-pytorch-distributed-quickstart.md b/docs/readthedocs/source/doc/Orca/QuickStart/orca-pytorch-distributed-quickstart.md
index 2dbf38d2fc6..3a20bfebbbe 100644
--- a/docs/readthedocs/source/doc/Orca/QuickStart/orca-pytorch-distributed-quickstart.md
+++ b/docs/readthedocs/source/doc/Orca/QuickStart/orca-pytorch-distributed-quickstart.md
@@ -130,4 +130,18 @@ for r in result:
     print(r, ":", result[r])
 ```
 
+### **Step 5: Save and Load the Model**
+
+Save the Estimator states (including model and optimizer) to the provided model path.
+
+```python
+est.save("mnist_model")
+```
+
+Load the Estimator states (model and possibly with optimizer) from provided model path. 
+
+```python
+est.load("mnist_model") 
+```
+
 **Note:** You should call `stop_orca_context()` when your application finishes.
diff --git a/docs/readthedocs/source/doc/Orca/QuickStart/orca-pytorch-quickstart.md b/docs/readthedocs/source/doc/Orca/QuickStart/orca-pytorch-quickstart.md
index 73bad54af7f..0bfce47e7de 100644
--- a/docs/readthedocs/source/doc/Orca/QuickStart/orca-pytorch-quickstart.md
+++ b/docs/readthedocs/source/doc/Orca/QuickStart/orca-pytorch-quickstart.md
@@ -132,4 +132,18 @@ for r in result:
     print(r, ":", result[r])
 ```
 
+### **Step 5: Save and Load the Model**
+
+Save the Estimator states (including model and optimizer) to the provided model path.
+
+```python
+est.save("mnist_model")
+```
+
+Load the Estimator states (model and possibly with optimizer) from the provided model path.
+
+```python
+est.load("mnist_model")
+```
+
 **Note:** You should call `stop_orca_context()` when your application finishes.
diff --git a/docs/readthedocs/source/doc/Orca/QuickStart/orca-tf2keras-quickstart.md b/docs/readthedocs/source/doc/Orca/QuickStart/orca-tf2keras-quickstart.md
index c460ef674d4..c1fd8f136fa 100644
--- a/docs/readthedocs/source/doc/Orca/QuickStart/orca-tf2keras-quickstart.md
+++ b/docs/readthedocs/source/doc/Orca/QuickStart/orca-tf2keras-quickstart.md
@@ -130,20 +130,20 @@ You could also save the model to Keras H5 format by passing `save_format='h5'`
 
 ```python
 # save model in SavedModel format
-estimator.save("/tmp/cifar10_model")
+est.save("/tmp/cifar10_model")
 
 # load model
-estimator.load("/tmp/cifar10_model")
+est.load("/tmp/cifar10_model")
 ```
 
 **2. HDF5 format**
 
 ```python
 # save model in H5 format
-estimator.save("/tmp/cifar10_model.h5", save_format='h5')
+est.save("/tmp/cifar10_model.h5", save_format='h5')
 
 # load model
-estimator.load("/tmp/cifar10_model.h5")
+est.load("/tmp/cifar10_model.h5")
 ```
 
 That's it, the same code can run seamlessly in your local laptop and to distribute K8s or Hadoop cluster.