未验证 提交 db876ac8 编写于 作者: S SunAhong1993 提交者: GitHub

Update pytorch2paddle.md

上级 2c01f3f7
......@@ -14,7 +14,7 @@ treelib
## 使用方式
```
``` python
from x2paddle.convert import pytorch2paddle
pytorch2paddle(module=torch_module,
save_dir="./pd_model",
......@@ -31,7 +31,7 @@ pytorch2paddle(module=torch_module,
## 使用示例
```
``` python
import torch
import numpy as np
from torchvision.models import AlexNet
......@@ -50,4 +50,4 @@ pytorch2paddle(torch_model,
save_dir="pd_model_trace",
jit_type="trace",
input_examples=[torch.tensor(input_data)])
```
\ No newline at end of file
```
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册