Torch | Paddle |
torch.cat() | paddle.concat() |
torch.clamp() | paddle.clip() |
x.shape | paddle.shape(x) |
nn.AdaptiveAvgPool2d | nn.AdaptiveAvgPool2D |
AvgPool2d | AvgPool2D |
conv2d | conv2D |
BatchNorm2d | BatchNorm2D |
nn.MaxPool2d | nn.MaxPool2D |
bias | bias_attr |
nn.ReLU(inplace=True) | nn.ReLU() |
nn.Module.modules | nn.Layer.sublayers |
- 全局搜索paddle 替换为 torch
- 全局搜索Layer 替换为 Module
- 类似Conv2D中大“D”更换为小“d”
- 修改axis 为dim
- 删除算子中的权重初始化变量:param_attr=ParamAttr(regularizer=L2Decay(0.0)),
bias_attr=ParamAttr(regularizer=L2Decay(0.0)) - paddle-picodet"torch化"记录 - 知乎
更多推荐
Paddle框架转Torch遇到的函数
发布评论