README.md 560 字节
Newer Older
Z
Zeyu Chen 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22
# Transformer Benchmark with Fleet API

### 静态图
如果是需要单机多卡训练,则使用下面的命令进行训练:
``` shell
cd static/
export CUDA_VISIBLE_DEVICES=0
python3 train.py
```

### 动态图
如果使用单机单卡进行训练可以使用如下命令:
``` shell
cd dygraph/
export CUDA_VISIBLE_DEVICES=0
python3 train.py
```

如果使用单机多卡进行训练可以使用如下命令:
``` shell
cd dygraph/
export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
23
python3 -m paddle.distributed.launch --gpus "0,1,2,3,4,5,6,7" train.py
Z
Zeyu Chen 已提交
24
```