seiriosplus/fleetrec

Extremely easy for distributed training

快速开始

环境准备

Fleet-Rec是基于飞桨分布式训练所开发的，包含模型、训练模式的快速开发、调试、部署的工具， 让用户更轻松的使用飞桨分布式训练。

• 安装方法一<PIP源直接安装>：

python -m pip install fleet-rec

• 安装方法二

• 安装飞桨 注：需要用户安装最新版本的飞桨<当前只支持Linux系统>。

python -m pip install paddlepaddle -i https://mirror.baidu.com/pypi/simple

• 源码安装Fleet-Rec

git clone https://github.com/seiriosPlus/FleetRec/
cd FleetRec
python setup.py install

ctr-dnn示例使用

目前框架内置了多个模型，简单的命令即可使用内置模型开始单机训练和本地1*1模拟训练

单机训练

cd FleetRec

python -m fleetrec.run \
       -m fleetrec.models.rank.dnn \
       -d cpu \
       -e single 

# 使用GPU资源进行训练
python -m fleetrec.run \
       -m fleetrec.models.rank.dnn \
       -d gpu \
       -e single

本地模拟分布式训练

cd FleetRec
# 使用CPU资源进行训练
python -m fleetrec.run \
       -m fleetrec.models.rank.dnn \
       -d cpu \
       -e local_cluster

集群提交分布式训练<需要用户预先配置好集群环境，本提交命令不包含提交客户端>

cd FleetRec

python -m fleetrec.run \
       -m fleetrec.models.rank.dnn \
       -d cpu \
       -e cluster

更多用户文档及二次开发文档，敬请期待。