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

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