SqueezeNet1_0.sh 329 字节
Newer Older
R
ruri 已提交
1 2 3 4 5 6 7 8 9 10 11 12
#SqueezeNet1_0
python train.py \
        --model=SqueezeNet1_0 \
        --batch_size=256 \
        --total_images=1281167 \
        --image_shape=3,224,224 \
        --lr_strategy=cosine_decay \
        --class_dim=1000 \
        --model_save_dir=output/ \
        --lr=0.02 \
        --num_epochs=120 \
        --l2_decay=1e-4