升级指南

    • 动态图功能完善,动态图模下数据表示概念为Tensor,推荐使用动态图模式;

    • API目录体系调整,API的命名和别名进行了统一规范化,虽然兼容老版API,但请使用新API体系开发;

    • 数据处理、组网方式、模型训练、多卡启动、模型保存和推理等开发流程都有了对应优化,请对应查看说明;

    以上变化请仔细阅读本指南。对于已有模型的升级,我们还提供了2.0转换工具(见附录)提供更自动化的辅助。 其他一些功能增加方面诸如动态图对量化训练、混合精度的支持、动静转换等方面不在本指南列出,具体可查看或对应文档。

    飞桨2.0版本将会把动态图作为默认模式(如果还想使用静态图,可通过调用切换)。

    使用Tensor概念表示数据

    静态图模式下,由于组网时使用的数据不能实时访问,Paddle用Variable来表示数据。 动态图下,从直观性等角度考虑,将数据表示概念统一为Tensor。动态图下Tensor的创建主要有两种方法:

    1. 通过调用paddle.to_tensor函数,将python scalar/list,或者numpy.ndarray数据转换为Paddle的Tensor。具体使用方法,请查看官网的API文档。
    1. import paddle
    2. import numpy as np
    3. paddle.to_tensor(1)
    4. paddle.to_tensor((1.1, 2.2))
    5. paddle.to_tensor(np.random.randn(3, 4))
    1. 通过调用 paddle.zeros, paddle.ones, paddle.full, paddle.arange, paddle.rand, paddle.randn, paddle.randint, paddle.normal, paddle.uniform 等函数,创建并返回Tensor。

    API目录结构

    为了API组织更加简洁和清晰,将原来padddle.fluid.xxx的目录体系全新升级为paddle.xxx,并对子目录的组织进行了系统的条理化优化。同时还增加了高层API,可以高低搭配使用。paddle.fluid目录下暂时保留了1.8版本API,主要是兼容性考虑,未来会被删除。 基于2.0的开发任务,请使用paddle目录下的API,不要再使用paddle.fluid目录下的API。 如果发现Paddle目录下有API缺失的情况,推荐使用基础API进行组合实现;您也可以通过在 github 上提issue的方式向我们反馈。

    2.0版本的API 整体目录结构如下

    API别名规则

    • 为了方便用户使用,API会在不同的路径下建立别名:

      • 所有device, framework, tensor目录下的API,均在paddle根目录建立别名;除少数特殊API外,其他API在paddle根目录下均没有别名。

      • paddle.nn目录下除functional目录以外的所有API,在paddle.nn目录下均有别名;functional目录中的API,在paddle.nn目录下均没有别名。

    • 推荐用户优先使用较短的路径的别名,比如paddle.add -> paddle.tensor.add,推荐优先使用paddle.add

    • 以下为一些特殊的别名关系,推荐使用左边的API名称:

      • paddle.tanh -> paddle.tensor.tanh -> paddle.nn.functional.tanh

      • paddle.remainder -> paddle.mod -> paddle.floor_mod

      • paddle.randn -> paddle.standard_normal

      • Layer.set_state_dict -> Layer.set_dict

    • 加、减、乘、除使用全称,不使用简称

    • 对于当前逐元素操作,不加elementwise前缀

    • 对于按照某一轴操作,不加reduce前缀

    • Conv, Pool, Dropout, BatchNorm, Pad组网类API根据输入数据类型增加1D, 2D, 3D后缀

    数据处理

    数据处理推荐使用paddle.io目录下的Dataset,Sampler, BatchSampler, DataLoader接口,不推荐reader类接口。一些常用的数据集已经在paddle.vision.datasets和paddle.text.datasets目录实现,具体参考API文档。

    1. from paddle.io import Dataset
    2. class MyDataset(Dataset):
    3. """
    4. 步骤一:继承paddle.io.Dataset类
    5. """
    6. def __init__(self, mode='train'):
    7. """
    8. 步骤二:实现构造函数,定义数据读取方式,划分训练和测试数据集
    9. """
    10. super(MyDataset, self).__init__()
    11. if mode == 'train':
    12. self.data = [
    13. ['traindata1', 'label1'],
    14. ['traindata2', 'label2'],
    15. ['traindata3', 'label3'],
    16. ['traindata4', 'label4'],
    17. ]
    18. else:
    19. self.data = [
    20. ['testdata1', 'label1'],
    21. ['testdata2', 'label2'],
    22. ['testdata3', 'label3'],
    23. ['testdata4', 'label4'],
    24. ]
    25. def __getitem__(self, index):
    26. """
    27. 步骤三:实现__getitem__方法,定义指定index时如何获取数据,并返回单条数据(训练数据,对应的标签)
    28. """
    29. data = self.data[index][0]
    30. label = self.data[index][1]
    31. return data, label
    32. def __len__(self):
    33. """
    34. 步骤四:实现__len__方法,返回数据集总数目
    35. """
    36. return len(self.data)
    37. # 测试定义的数据集
    38. val_dataset = MyDataset(mode='test')
    39. print('=============train dataset=============')
    40. for data, label in train_dataset:
    41. print(data, label)
    42. print('=============evaluation dataset=============')
    43. for data, label in val_dataset:
    44. print(data, label)

    组网方式

    Sequential 组网

    针对顺序的线性网络结构我们可以直接使用Sequential来快速完成组网,可以减少类的定义等代码编写。

    1. import paddle
    2. # Sequential形式组网
    3. mnist = paddle.nn.Sequential(
    4. paddle.nn.Flatten(),
    5. paddle.nn.Linear(784, 512),
    6. paddle.nn.ReLU(),
    7. paddle.nn.Dropout(0.2),
    8. paddle.nn.Linear(512, 10)
    9. )

    SubClass组网

    针对一些比较复杂的网络结构,就可以使用Layer子类定义的方式来进行模型代码编写,在__init__构造函数中进行组网Layer的声明,在forward中使用声明的Layer变量进行前向计算。子类组网方式也可以实现sublayer的复用,针对相同的layer可以在构造函数中一次性定义,在forward中多次调用。

    模型训练

    使用高层API

    增加了paddle.Model高层API,大部分任务可以使用此API用于简化训练、评估、预测类代码开发。注意区别Model和Net概念,Net是指继承paddle.nn.Layer的网络结构;而Model是指持有一个Net对象,同时指定损失函数、优化算法、评估指标的可训练、评估、预测的实例。具体参考高层API的代码示例。

    1. import paddle
    2. from paddle.vision.transforms import ToTensor
    3. train_dataset = paddle.vision.datasets.MNIST(mode='train', transform=ToTensor())
    4. test_dataset = paddle.vision.datasets.MNIST(mode='test', transform=ToTensor())
    5. lenet = paddle.vision.models.LeNet()
    6. # Mnist继承paddle.nn.Layer属于Net,model包含了训练功能
    7. model = paddle.Model(lenet)
    8. # 设置训练模型所需的optimizer, loss, metric
    9. model.prepare(
    10. paddle.optimizer.Adam(learning_rate=0.001, parameters=model.parameters()),
    11. paddle.nn.CrossEntropyLoss(),
    12. paddle.metric.Accuracy(topk=(1, 2))
    13. )
    14. # 启动训练
    15. # 启动评估
    16. model.evaluate(test_dataset, log_freq=20, batch_size=64)

    使用基础API

    1. import paddle
    2. from paddle.vision.transforms import ToTensor
    3. train_dataset = paddle.vision.datasets.MNIST(mode='train', transform=ToTensor())
    4. test_dataset = paddle.vision.datasets.MNIST(mode='test', transform=ToTensor())
    5. lenet = paddle.vision.models.LeNet()
    6. loss_fn = paddle.nn.CrossEntropyLoss()
    7. # 加载训练集 batch_size 设为 64
    8. train_loader = paddle.io.DataLoader(train_dataset, batch_size=64, shuffle=True)
    9. def train():
    10. epochs = 2
    11. adam = paddle.optimizer.Adam(learning_rate=0.001, parameters=lenet.parameters())
    12. # 用Adam作为优化函数
    13. for epoch in range(epochs):
    14. for batch_id, data in enumerate(train_loader()):
    15. x_data = data[0]
    16. y_data = data[1]
    17. predicts = lenet(x_data)
    18. acc = paddle.metric.accuracy(predicts, y_data)
    19. loss = loss_fn(predicts, y_data)
    20. loss.backward()
    21. if batch_id % 100 == 0:
    22. print("epoch: {}, batch_id: {}, loss is: {}, acc is: {}".format(epoch, batch_id, loss.numpy(), acc.numpy()))
    23. adam.step()
    24. adam.clear_grad()
    25. # 启动训练
    26. train()

    2.0增加paddle.distributed.spawn函数来启动单机多卡训练,同时原有的paddle.distributed.launch的方式依然保留。

    方式1、launch启动

    高层API场景

    当调用paddle.Model高层来实现训练时,想要启动单机多卡训练非常简单,代码不需要做任何修改,只需要在启动时增加一下参数-m paddle.distributed.launch

    1. # 单机单卡启动,默认使用第0号卡
    2. $ python train.py
    3. # 单机多卡启动,默认使用当前可见的所有卡
    4. $ python -m paddle.distributed.launch train.py
    5. # 单机多卡启动,设置当前使用的第0号和第1号卡
    6. $ python -m paddle.distributed.launch --selected_gpus='0,1' train.py
    7. # 单机多卡启动,设置当前使用第0号和第1号卡
    8. $ export CUDA_VISIBLE_DEVICES=0,1
    9. $ python -m paddle.distributed.launch train.py
    基础API场景

    如果使用基础API实现训练,想要启动单机多卡训练,需要对单机单卡的代码进行3处修改,具体如下:

    修改完后保存文件,然后使用跟高层API相同的启动方式即可

    注意: 单卡训练不支持调用 init_parallel_env,请使用以下几种方式进行分布式训练。

    1. # 单机多卡启动,默认使用当前可见的所有卡
    2. $ python -m paddle.distributed.launch train.py
    3. # 单机多卡启动,设置当前使用的第0号和第1号卡
    4. $ python -m paddle.distributed.launch --selected_gpus '0,1' train.py
    5. # 单机多卡启动,设置当前使用第0号和第1号卡
    6. $ export CUDA_VISIBLE_DEVICES=0,1
    7. $ python -m paddle.distributed.launch train.py

    方式2、spawn启动

    1. from __future__ import print_function
    2. import paddle
    3. import paddle.nn as nn
    4. import paddle.optimizer as opt
    5. import paddle.distributed as dist
    6. class LinearNet(nn.Layer):
    7. def __init__(self):
    8. super(LinearNet, self).__init__()
    9. self._linear1 = nn.Linear(10, 10)
    10. self._linear2 = nn.Linear(10, 1)
    11. def forward(self, x):
    12. return self._linear2(self._linear1(x))
    13. def train(print_result=False):
    14. # 1. 初始化并行训练环境
    15. dist.init_parallel_env()
    16. # 2. 创建并行训练 Layer 和 Optimizer
    17. layer = LinearNet()
    18. dp_layer = paddle.DataParallel(layer)
    19. loss_fn = nn.MSELoss()
    20. adam = opt.Adam(
    21. learning_rate=0.001, parameters=dp_layer.parameters())
    22. # 3. 运行网络
    23. inputs = paddle.randn([10, 10], 'float32')
    24. outputs = dp_layer(inputs)
    25. labels = paddle.randn([10, 1], 'float32')
    26. loss = loss_fn(outputs, labels)
    27. if print_result is True:
    28. print("loss:", loss.numpy())
    29. loss.backward()
    30. adam.step()
    31. adam.clear_grad()
    32. # 使用方式1:仅传入训练函数
    33. # 适用场景:训练函数不需要任何参数,并且需要使用所有当前可见的GPU设备并行训练
    34. if __name__ == '__main__':
    35. dist.spawn(train)
    36. # 使用方式2:传入训练函数和参数
    37. # 适用场景:训练函数需要一些参数,并且需要使用所有当前可见的GPU设备并行训练
    38. if __name__ == '__main__':
    39. dist.spawn(train, args=(True,))
    40. # 使用方式3:传入训练函数、参数并指定并行进程数
    41. # 适用场景:训练函数需要一些参数,并且仅需要使用部分可见的GPU设备并行训练,例如:
    42. # 当前机器有8张GPU卡 {0,1,2,3,4,5,6,7},此时会使用前两张卡 {0,1};
    43. # 或者当前机器通过配置环境变量 CUDA_VISIBLE_DEVICES=4,5,6,7,仅使4张
    44. # GPU卡可见,此时会使用可见的前两张卡 {4,5}
    45. if __name__ == '__main__':
    46. dist.spawn(train, args=(True,), nprocs=2)
    47. # 使用方式4:传入训练函数、参数、指定进程数并指定当前使用的卡号
    48. # 使用场景:训练函数需要一些参数,并且仅需要使用部分可见的GPU设备并行训练,但是
    49. # 可能由于权限问题,无权配置当前机器的环境变量,例如:当前机器有8张GPU卡
    50. # {0,1,2,3,4,5,6,7},但你无权配置CUDA_VISIBLE_DEVICES,此时可以通过
    51. # 指定参数 selected_gpus 选择希望使用的卡,例如 selected_gpus='4,5',
    52. # 可以指定使用第4号卡和第5号卡
    53. if __name__ == '__main__':
    54. dist.spawn(train, nprocs=2, selected_gpus='4,5')
    55. # 使用方式5:指定多卡通信的起始端口
    56. # 使用场景:端口建立通信时提示需要重试或者通信建立失败
    57. # Paddle默认会通过在当前机器上寻找空闲的端口用于多卡通信,但当机器使用环境
    58. # 较为复杂时,程序找到的端口可能不够稳定,此时可以自行指定稳定的空闲起始
    59. # 端口以获得更稳定的训练体验
    60. if __name__ == '__main__':
    61. dist.spawn(train, nprocs=2, started_port=12345)

    模型保存

    Paddle保存的模型有两种格式,一种是训练格式,保存模型参数和优化器相关的状态,可用于恢复训练;一种是预测格式,保存预测的静态图网络结构以及参数,用于预测部署。

    高层API场景

    高层API下用于预测部署的模型保存方法为:

    1. model = paddle.Model(Mnist())
    2. # 预测格式,保存的模型可用于预测部署
    3. model.save('mnist', training=False)
    4. # 保存后可以得到预测部署所需要的模型

    基础API场景

    动态图训练的模型,可以通过动静转换功能,转换为可部署的静态图模型,具体做法如下:

    推理

    推理库Paddle Inference的API做了升级,简化了写法,以及去掉了历史上冗余的概念。API的变化为纯增,原有API保持不变,但推荐新的API体系,旧API在后续版本会逐步删除。

    C++ API

    重要变化:

    • 命名空间从 paddle 变更为 paddle_infer

    • PaddleTensor, PaddleBuf 等被废弃,ZeroCopyTensor 变为默认 Tensor 类型,并更名为 Tensor

    • 新增 PredictorPool 工具类简化多线程 predictor 的创建,后续也会增加更多周边工具

    • CreatePredictor (原 CreatePaddlePredictor) 的返回值由 unique_ptr 变为 shared_ptr 以避免 Clone 后析构顺序出错的问题

    API 变更

    使用新 C++ API 的流程与之前完全一致,只有命名变化

    1. #include "paddle_infernce_api.h"
    2. using namespace paddle_infer;
    3. Config config;
    4. config.SetModel("xxx_model_dir");
    5. auto predictor = CreatePredictor(config);
    6. // Get the handles for the inputs and outputs of the model
    7. auto input0 = predictor->GetInputHandle("X");
    8. auto output0 = predictor->GetOutputHandle("Out");
    9. for (...) {
    10. // Assign data to input0
    11. MyServiceSetData(input0);
    12. predictor->Run();
    13. // get data from the output0 handle

    Python API

    Python API 的变更与 C++ 基本对应,会在2.0版发布。

    2.0转换工具

    为了降级代码升级的成本,我们提供了转换工具,可以帮助将Paddle 1.8版本开发的代码,升级为2.0的API。由于相比于Paddle 1.8版本,2.0版本的API进行了大量的升级,包括API名称,参数名称,行为等。转换工具当前还不能覆盖所有的API升级;对于无法转换的API,转换工具会报错,提示用户手动升级。

    对于转换工具没有覆盖的API,请查看官网的API文档,手动升级代码的API。

    您可以在官网应用实践栏目内进行在线浏览,也可以下载在这里提供的源代码: