架构重构:
- 根据任务类型:重构了 plsc 整个架构,核心的训练、离线评测、模型导出根据任务封装成 engine。
- 子模块划分:根据训练和推理的环节,设计子模块,抽取到 plsc 目录中。
- 工作空间解耦:具体任务的工作空间和 plsc 库解耦,每个模型在 task 目录下拥有独立的目录
- 测试覆盖:新增了 CI 和 CE 功能,CI 覆盖 PR 级别的测试,CE 覆盖周级别的回归测试。
新增模型:
- 人脸识别:
- 新增 FaceViT,发布 3 个 checkpoint
- 新增 MobileFaceNet_base,发布 1 个 checkpoint
- 有监督大模型预训练:
- 新增 ViT,发布 4 个 checkpoint;
- 新增 DeiT,发布 3 个 checkpoint;
- 新增 CaiT,发布 1 个 checkpoint;
- 自监督大模型预训练:
- 新增 MAE,发布 3 个 checkpoint;
- 新增 MoCo v3,发布 1 个 checkpoint;
新增功能:
- PIL 数据增广
- ImageFolder 数据集
Bug修复:
- ParitalFC 采样和梯度稀疏更新功能和精度修复
文档更新:
- 部分数据增广的字段说明
- Configuration 的字段说明
PaddlePaddle 版本:paddlepaddle-gpu>=2.4.0,<=2.5.0