Torch                Paddle
torch.cat()paddle.concat()
torch.clamp()paddle.clip()
x.shapepaddle.shape(x) 
nn.AdaptiveAvgPool2dnn.AdaptiveAvgPool2D
AvgPool2dAvgPool2D
conv2d conv2D
BatchNorm2dBatchNorm2D
nn.MaxPool2dnn.MaxPool2D
biasbias_attr
nn.ReLU(inplace=True)nn.ReLU()
nn.Module.modulesnn.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遇到的函数