paddle高层API学习笔记（四）模型搭建的三种方式

在飞桨框架中，针对用户不同场景，有三种方式来构建模型

1、 Sequentia形式组网

# 1、 Sequentia形式组网

mnist = paddle.nn.Sequential(
    paddle.nn.Flatten(),
    paddle.nn.Linear(784, 512),
    paddle.nn.ReLU(),
    paddle.nn.Linear(512, 10)
)

2、 Subclass形式组网

# 2、 Subclass形式组网
class Mnist(paddle.nn.Layer):
    def __init__(self):
        super(Mnist, self).__init__()

        self.flatten = paddle.nn.Flatten()
        self.linear_1 = paddle.nn.Linear(784, 512)
        self.linear_2 = paddle.nn.Linear(512, 10)
        self.relu = paddle.nn.ReLU()

    def forward(self, inputs):
        y = self.flatten(inputs)
        y = self.linear_1(y)
        y = self.relu(y)
        y = self.linear_2(y)

        return y

3、框架内置模型

# 3、框架内置模型
lenet = paddle.vision.models.LeNet()

深度学习万能公式