mobilenet_v1_x1.0.sh 418 字节
Newer Older
L
LielinJiang 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13
export CUDA_VISIBLE_DEVICES=0,1,2,3 

# 默认imagenet数据存储在data/ILSVRC2012/下,去除-d便使用静态图模式运行
python -m paddle.distributed.launch main.py \
        --arch mobilenet_v1 \
        --epoch 120 \
        --batch-size 64 \
        --learning-rate 0.1 \
        --lr-scheduler piecewise \
        --milestones 30 60 90 \
        --weight-decay 3e-5 \
        -d \
        data/ILSVRC2012/