kuma202204141717325/examples/panguAlpha_pytorch

• 代码修改
• 环境
• AISynergy 的使用
• Pytorch 框架下的同构训练
• • • 单卡训练，数据并行，模型并行
    • 1、运行 server 端
    • 2、运行 client 端
• Pytorch 和 MindSpore 框架下的异构训练

只需要两行代码就可以轻松的把 Pangu-alpha 扩展成多集群协同训练！！！

本例子已完美支持 350M 和 2.6B 模型

本例子实现了 Pangu-alpha 模型在 AISynergy 平台上进行协同训练，并支持 Pytorch 框架下的数据并行和模型并行的协同训练，也支持 Pytorch 和 MindSpore 异构框架下的协同训练。

代码修改

下载Pangu-alpha代码

1、只需要在文件 megatron/training.py 增加下面的代码片段

client = PanguClient(args, model, train, forward_step_func,
                     optimizer, lr_scheduler, train_data_iterator, valid_data_iterator,
                     evaluate_and_print_results, model_provider, get_timers,
                     train_iters_per_round=20)
client.run("localhost:6070", 2**31 -1)

2、接管 megatron.mpu 模块，详情查看文件megatron/mpu/__init__.py

环境

支持 python >= 3.6, pytorch >= 1.5, cuda >= 10, and nccl >= 2.6 版本.

推荐使用英伟达的官方 docker 镜像docker pull nvcr.io/nvidia/pytorch:20.03-py3。
需要安装 NLTK.

也可直接下载配好的镜像：

docker pull yands/pangu-alpha-megatron-lm-nvidia-pytorch:20.03.2

激活环境：source /opt/conda/bin/activate。

根据教程部署下载和安装 AISynergy

AISynergy 的使用

AISynergy提供了两种使用方式：网页前端启动任务和服务器手动启动任务，两种方式选其一

1、基于网页前端启动任务，可以参考文档平台使用手册。

2、基于服务器手动启动任务，可以参考下面步骤。

Pytorch 框架下的同构训练

请先阅读Pangu-alpha项目下的 pytorch 版本的训练步骤和数据生成步骤。

单卡训练，数据并行，模型并行

协同训练已适配 Pangu-alpha 的所有并行模式，只需要配置相关参数即可

设置参数：

1、模型相关的具体参数可以参考项目Pangu-alpha

2、train_iters_per_round: 设置每轮的迭代次数 (training.py)

3、IP:Port: 设置 server 的 ip 和端口 (server.py, training.py)

4、num_rounds: 设置训练轮数(server.py)

1、运行 server 端

运行脚本python server.py

2、运行 client 端

1、修改运行脚本
不同的并行策略只需要修改脚本pretrain_pangu_AISynergy_350m.sh或pretrain_pangu_AISynergy_2.6b.sh

单卡训练:
    GPUS_PER_NODE=1

数据并行:
    GPUS_PER_NODE=2
    --model-parallel-size=1

2路数据并行，2路模型并行:
    GPUS_PER_NODE = 4
    --model-parallel-size=2

2、运行脚本：

如果在同一机器内运行不同的 client 需要指定不同的 nccl 端口，否则会冲突。

如果是 client 运行在不同机器上，端口设置成一样就行。

# 350M
bash pretrain_pangu_AISynergy_350m.sh 6060 
bash pretrain_pangu_AISynergy_350m.sh 6061

#2.6B
bash pretrain_pangu_AISynergy_2.6B.sh 6060 
bash pretrain_pangu_AISynergy_2.6B.sh 6061

Pytorch 和 MindSpore 框架下的异构训练

本项目已经完成了适配 MindSpore 框架的的工作：

1、Pytorch 端的起任务的步骤请查看上一节： Pytorch 框架下的同构训练

2、需要同时启动 MindSpore 版本的 client，请参考例子pangu-mindspore