resnet152.sh 317 字节
Newer Older
L
LielinJiang 已提交
1 2 3 4 5 6 7 8 9 10
export CUDA_VISIBLE_DEVICES=0,1,2,3 

# 默认imagenet数据存储在data/ILSVRC2012/下,去除-d便使用静态图模式运行
python -m paddle.distributed.launch main.py \
        --arch resnet152 \
        --epoch 90 \
        --batch-size 64 \
        --learning-rate 0.1 \
        -d \
        data/ILSVRC2012/