ShuffleNetV2_x2_0.sh 336 字节
Newer Older
R
ruri 已提交
1 2 3 4 5 6 7 8 9 10 11 12
#ShuffleNetV2_x2_0:
python train.py \
       --model=ShuffleNetV2_x2_0 \
       --batch_size=512 \
       --total_images=1281167 \
       --class_dim=1000 \
       --image_shape=3,224,224 \
       --model_save_dir=output/ \
       --lr_strategy=cosine_decay_warmup \
       --num_epochs=240 \
       --lr=0.25 \
       --l2_decay=4e-5