Merge pull request #732 from wjj19950828/Fixed_pytorch_readme

Add PyTorch2Paddle Dynamic shape export Readme

docs/inference_model_convertor/pytorch2paddle.md

@@ -14,7 +14,7 @@ treelib
 
 ## 使用方式
 
-``` python
+```python
 from x2paddle.convert import pytorch2paddle
 pytorch2paddle(module=torch_module,
                save_dir="./pd_model",
@@ -26,12 +26,15 @@ pytorch2paddle(module=torch_module,
 # input_examples (list[torch.tensor]): torch.nn.Module的输入示例，list的长度必须与输入的长度一致。默认为None。
 ```
 
-**注意：** 当jit_type为"trace"时，input_examples不可为None，转换后自动进行动转静；
-          当jit_type为"script"时"，input_examples不为None时，才可以进行动转静。
+**注意:**
+- jit_type为"trace"时，input_examples不可为None，转换后自动进行动转静；
+- jit_type为"script"时，当input_examples为None时，只生成动态图代码；当input_examples不为None时，才能自动动转静。
 
 ## 使用示例
 
-``` python
+### Trace 模式
+
+```python
 import torch
 import numpy as np
 from torchvision.models import AlexNet
@@ -51,3 +54,61 @@ pytorch2paddle(torch_module,
                jit_type="trace",
                input_examples=[torch.tensor(input_data)])
 ```
+
+### 动态 shape 导出
+
+#### 方式一：PyTorch->ONNX->Paddle
+
+```python
+import torch
+from torchvision.models import AlexNet
+from torchvision.models.utils import load_state_dict_from_url
+
+# 获取PyTorch Module
+torch_module = AlexNet()
+torch_state_dict = load_state_dict_from_url('https://download.pytorch.org/models/alexnet-owt-4df8aa71.pth')
+torch_module.load_state_dict(torch_state_dict)
+# 设置为eval模式
+torch_module.eval()
+input_names = ["input_0"]
+output_names = ["output_0"]
+
+x = torch.randn((1, 3, 224, 224))
+y = torch.randn((1, 1000))
+
+torch.onnx.export(torch_module, x, 'model.onnx', opset_version=11, input_names=input_names,
+                  output_names=output_names, dynamic_axes={'input_0': [0], 'output_0': [0]})
+```
+
+导出 ONNX 动态 shape 模型，更多细节参考[相关文档](https://pytorch.org/docs/stable/onnx.html?highlight=onnx%20export#torch.onnx.export)
+
+然后通过 X2Paddle 命令导出 Paddle 模型
+
+```shell
+x2paddle --framework=onnx --model=model.onnx --save_dir=pd_model_dynamic
+```
+
+#### 方式二：手动动转静
+
+在自动生成的 x2paddle_code.py 中添加如下代码：
+
+```python
+def main(x0):
+    # There are 0 inputs.
+    paddle.disable_static()
+    params = paddle.load('model.pdparams')
+    model = AlexNet()
+    model.set_dict(params)
+    model.eval()
+    ## convert to jit
+    sepc_list = list()
+    sepc_list.append(
+            paddle.static.InputSpec(
+                shape=[-1, 3, 224, 224], name="x0", dtype="float32"))
+    static_model = paddle.jit.to_static(model, input_spec=sepc_list)
+    paddle.jit.save(static_model, "pd_model_trace/inference_model/model")
+    out = model(x0)
+    return out
+```
+
+然后运行 main 函数导出动态 shape 模型