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

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