关于paddle定义神经网络跟pytorch的转
收藏
比如pytorch有专门的nn.ReLU(True)以及nn.ReflectionPad2d(3)这种对象,
paddle中是函数形式调用fluid.layers.pad2d(imput,paddings=1)以及paddle.fluid.layers.leaky_relu(x, alpha=0.2),无法单独定义一层网络,
有没有什么方法可以跟pytorch一样方便?
0
收藏
请登录后评论
可以自己封装一下呀
用动态图写一个类继承fluid.dygraph.Layer,然后实现init和forward方法,然后后面调用就是一层了
是的,我自己封装了一下,只能自己写,目前版本还没有对象,只有函数