飞桨框架IPU版训练示例

BERT-Base训练示例

示例将默认用户已安装飞桨框架IPU版,并且已经配置运行时需要的环境(建议在Docker环境中使用飞桨框架IPU版)。

示例代码位于 Paddle-BERT with Graphcore IPUs

第一步:下载源码并安装依赖

# 下载源码
git clone https://github.com/PaddlePaddle/PaddleNLP.git
cd model_zoo/bert/static_ipu/

# 安装依赖
pip install -r requirements.txt

第二步:准备数据集

按照 README.md 的描述准备用于预训练的数据集。

第三步:执行模型训练

按照 README.md 的描述开始BERT-Base模型的预训练和在SQuAD v1.1数据集上的模型微调。