Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
X2Paddle
提交
e231eca2
X
X2Paddle
项目概览
PaddlePaddle
/
X2Paddle
8 个月 前同步成功
通知
327
Star
698
Fork
167
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
26
列表
看板
标记
里程碑
合并请求
4
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
X
X2Paddle
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
26
Issue
26
列表
看板
标记
里程碑
合并请求
4
合并请求
4
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
前往新版Gitcode,体验更适合开发者的 AI 搜索 >>
未验证
提交
e231eca2
编写于
6月 03, 2021
作者:
S
SunAhong1993
提交者:
GitHub
6月 03, 2021
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
修复针对2.0.1的paddle_type处理 (#612)
* Update stargan.md * fix the paddle_type * add docs
上级
9277f0bc
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
26 addition
and
27 deletion
+26
-27
docs/pytorch_project_convertor/API_docs/vision/torchvision.models.md
...h_project_convertor/API_docs/vision/torchvision.models.md
+17
-17
x2paddle/utils.py
x2paddle/utils.py
+9
-10
未找到文件。
docs/pytorch_project_convertor/API_docs/vision/torchvision.models.md
浏览文件 @
e231eca2
...
...
@@ -17,20 +17,20 @@ out = model(x)
目前支持的模型为:
| PyTorch模型 | Paddle模型 |
| ------------------------------------------------------------ | -------------------------------- |
|
[
torchvision.models.resnet18
](
https://pytorch.org/vision/stable/models.html#torchvision.models.resnet18
)
|
x2paddle.models.resnet18_pth
|
|
[
torchvision.models.resnet34
](
https://pytorch.org/vision/stable/models.html#torchvision.models.resnet34
)
|
x2paddle.models.resnet34_pth
|
|
[
torchvision.models.resnet50
](
https://pytorch.org/vision/stable/models.html#torchvision.models.resnet50
)
|
x2paddle.models.resnet50_pth
|
|
[
torchvision.models.resnet101
](
https://pytorch.org/vision/stable/models.html#torchvision.models.resnet101
)
|
x2paddle.models.resnet101_pth
|
|
[
torchvision.models.resnet152
](
https://pytorch.org/vision/stable/models.html#torchvision.models.resnet152
)
|
x2paddle.models.resnet152_pth
|
|
[
torchvision.models.resnext50_32x4d
](
https://pytorch.org/vision/stable/models.html#torchvision.models.resnext50_32x4d
)
|
x2paddle.models.resnext50_32x4d_pth
|
|
[
torchvision.models.resnext101_32x8d
](
https://pytorch.org/vision/stable/models.html#torchvision.models.resnext101_32x8d
)
|
x2paddle.resnext101_32x8d_pth
|
|
[
torchvision.models.wide_resnet50_2
](
https://pytorch.org/vision/stable/models.html#torchvision.models.wide_resnet50_2
)
|
x2paddle.models.wide_resnet50_2_pth
|
|
[
torchvision.models.wide_resnet101_2
](
https://pytorch.org/vision/stable/models.html#torchvision.models.wide_resnet101_2
)
|
x2paddle.models.wide_resnet101_2_pth
|
|
[
torchvision.models.vgg11
](
https://pytorch.org/vision/stable/models.html#torchvision.models.vgg11
)
|
x2paddle.models.vgg11_pth
|
|
[
torchvision.models.vgg11_bn
](
https://pytorch.org/vision/stable/models.html#torchvision.models.vgg11_bn
)
|
x2paddle.models.vgg11_bn_pth
|
|
[
torchvision.models.vgg13
](
https://pytorch.org/vision/stable/models.html#torchvision.models.vgg13
)
|
x2paddle.models.vgg13_pth
|
|
[
torchvision.models.vgg13_bn
](
https://pytorch.org/vision/stable/models.html#torchvision.models.vgg13_bn
)
|
x2paddle.models.vgg13_bn_pth
|
|
[
torchvision.models.vgg16
](
https://pytorch.org/vision/stable/models.html#torchvision.models.vgg16
)
|
x2paddle.models.vgg16_pth
|
|
[
torchvision.models.vgg16_bn
](
https://pytorch.org/vision/stable/models.html#torchvision.models.vgg16_bn
)
|
x2paddle.models.vgg16_bn_pth
|
|
[
torchvision.models.vgg19
](
https://pytorch.org/vision/stable/models.html#torchvision.models.vgg19
)
|
x2paddle.models.vgg19_pth
|
|
[
torchvision.models.vgg19_bn
](
https://pytorch.org/vision/stable/models.html#torchvision.models.vgg19_bn
)
|
x2paddle.models.vgg19_bn_pth
|
|
[
torchvision.models.resnet18
](
https://pytorch.org/vision/stable/models.html#torchvision.models.resnet18
)
|
[
x2paddle.models.resnet18_pth
](
https://github.com/PaddlePaddle/X2Paddle/blob/develop/x2paddle/project_convertor/pytorch/models/resnet.py#L310
)
|
|
[
torchvision.models.resnet34
](
https://pytorch.org/vision/stable/models.html#torchvision.models.resnet34
)
|
[
x2paddle.models.resnet34_pth
](
https://github.com/PaddlePaddle/X2Paddle/blob/develop/x2paddle/project_convertor/pytorch/models/resnet.py#L322
)
|
|
[
torchvision.models.resnet50
](
https://pytorch.org/vision/stable/models.html#torchvision.models.resnet50
)
|
[
x2paddle.models.resnet50_pth
](
https://github.com/PaddlePaddle/X2Paddle/blob/develop/x2paddle/project_convertor/pytorch/models/resnet.py#L334
)
|
|
[
torchvision.models.resnet101
](
https://pytorch.org/vision/stable/models.html#torchvision.models.resnet101
)
|
[
x2paddle.models.resnet101_pth
](
https://github.com/PaddlePaddle/X2Paddle/blob/develop/x2paddle/project_convertor/pytorch/models/resnet.py#L346
)
|
|
[
torchvision.models.resnet152
](
https://pytorch.org/vision/stable/models.html#torchvision.models.resnet152
)
|
[
x2paddle.models.resnet152_pth
](
https://github.com/PaddlePaddle/X2Paddle/blob/develop/x2paddle/project_convertor/pytorch/models/resnet.py#L358
)
|
|
[
torchvision.models.resnext50_32x4d
](
https://pytorch.org/vision/stable/models.html#torchvision.models.resnext50_32x4d
)
|
[
x2paddle.models.resnext50_32x4d_pth
](
https://github.com/PaddlePaddle/X2Paddle/blob/develop/x2paddle/project_convertor/pytorch/models/resnet.py#L370
)
|
|
[
torchvision.models.resnext101_32x8d
](
https://pytorch.org/vision/stable/models.html#torchvision.models.resnext101_32x8d
)
|
[
x2paddle.resnext101_32x8d_pth
](
https://github.com/PaddlePaddle/X2Paddle/blob/develop/x2paddle/project_convertor/pytorch/models/resnet.py#L385
)
|
|
[
torchvision.models.wide_resnet50_2
](
https://pytorch.org/vision/stable/models.html#torchvision.models.wide_resnet50_2
)
|
[
x2paddle.models.wide_resnet50_2_pth
](
https://github.com/PaddlePaddle/X2Paddle/blob/develop/x2paddle/project_convertor/pytorch/models/resnet.py#L400
)
|
|
[
torchvision.models.wide_resnet101_2
](
https://pytorch.org/vision/stable/models.html#torchvision.models.wide_resnet101_2
)
|
[
x2paddle.models.wide_resnet101_2_pth
](
https://github.com/PaddlePaddle/X2Paddle/blob/develop/x2paddle/project_convertor/pytorch/models/resnet.py#L419
)
|
|
[
torchvision.models.vgg11
](
https://pytorch.org/vision/stable/models.html#torchvision.models.vgg11
)
|
[
x2paddle.models.vgg11_pth
](
https://github.com/PaddlePaddle/X2Paddle/blob/develop/x2paddle/project_convertor/pytorch/models/vgg.py#L117
)
|
|
[
torchvision.models.vgg11_bn
](
https://pytorch.org/vision/stable/models.html#torchvision.models.vgg11_bn
)
|
[
x2paddle.models.vgg11_bn_pth
](
https://github.com/PaddlePaddle/X2Paddle/blob/develop/x2paddle/project_convertor/pytorch/models/vgg.py#L128
)
|
|
[
torchvision.models.vgg13
](
https://pytorch.org/vision/stable/models.html#torchvision.models.vgg13
)
|
[
x2paddle.models.vgg13_pth
](
https://github.com/PaddlePaddle/X2Paddle/blob/develop/x2paddle/project_convertor/pytorch/models/vgg.py#L139
)
|
|
[
torchvision.models.vgg13_bn
](
https://pytorch.org/vision/stable/models.html#torchvision.models.vgg13_bn
)
|
[
x2paddle.models.vgg13_bn_pth
](
https://github.com/PaddlePaddle/X2Paddle/blob/develop/x2paddle/project_convertor/pytorch/models/vgg.py#L150
)
|
|
[
torchvision.models.vgg16
](
https://pytorch.org/vision/stable/models.html#torchvision.models.vgg16
)
|
[
x2paddle.models.vgg16_pth
](
https://github.com/PaddlePaddle/X2Paddle/blob/develop/x2paddle/project_convertor/pytorch/models/vgg.py#L161
)
|
|
[
torchvision.models.vgg16_bn
](
https://pytorch.org/vision/stable/models.html#torchvision.models.vgg16_bn
)
|
[
x2paddle.models.vgg16_bn_pth
](
https://github.com/PaddlePaddle/X2Paddle/blob/develop/x2paddle/project_convertor/pytorch/models/vgg.py#L172
)
|
|
[
torchvision.models.vgg19
](
https://pytorch.org/vision/stable/models.html#torchvision.models.vgg19
)
|
[
x2paddle.models.vgg19_pth
](
https://github.com/PaddlePaddle/X2Paddle/blob/develop/x2paddle/project_convertor/pytorch/models/vgg.py#L183
)
|
|
[
torchvision.models.vgg19_bn
](
https://pytorch.org/vision/stable/models.html#torchvision.models.vgg19_bn
)
|
[
x2paddle.models.vgg19_bn_pth
](
https://github.com/PaddlePaddle/X2Paddle/blob/develop/x2paddle/project_convertor/pytorch/models/vgg.py#L194
)
|
x2paddle/utils.py
浏览文件 @
e231eca2
...
...
@@ -45,16 +45,15 @@ class PaddleDtypes():
self
.
t_int64
=
paddle
.
int64
self
.
t_bool
=
paddle
.
bool
else
:
from
paddle.fluid.core
import
VarDesc
self
.
t_float16
=
VarDesc
.
VarType
.
FP16
self
.
t_float32
=
VarDesc
.
VarType
.
FP32
self
.
t_float64
=
VarDesc
.
VarType
.
FP64
self
.
t_uint8
=
VarDesc
.
VarType
.
UINT8
self
.
t_int8
=
VarDesc
.
VarType
.
INT8
self
.
t_int16
=
VarDesc
.
VarType
.
INT16
self
.
t_int32
=
VarDesc
.
VarType
.
INT32
self
.
t_int64
=
VarDesc
.
VarType
.
INT64
self
.
t_bool
=
VarDesc
.
VarType
.
BOOL
self
.
t_float16
=
"paddle.fluid.core.VarDesc.VarType.FP16"
self
.
t_float32
=
"paddle.fluid.core.VarDesc.VarType.FP32"
self
.
t_float64
=
"paddle.fluid.core.VarDesc.VarType.FP64"
self
.
t_uint8
=
"paddle.fluid.core.VarDesc.VarType.UINT8"
self
.
t_int8
=
"paddle.fluid.core.VarDesc.VarType.INT8"
self
.
t_int16
=
"paddle.fluid.core.VarDesc.VarType.INT16"
self
.
t_int32
=
"paddle.fluid.core.VarDesc.VarType.INT32"
self
.
t_int64
=
"paddle.fluid.core.VarDesc.VarType.INT64"
self
.
t_bool
=
"paddle.fluid.core.VarDesc.VarType.BOOL"
is_new_version
=
check_version
()
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录