DarkNet53.sh 406 字节
Newer Older
R
ruri 已提交
1 2 3 4 5
#Training details
#DarkNet53
python train.py \
       --model=DarkNet53 \
       --batch_size=256 \
R
ruri 已提交
6
       --image_shape 3 256 256 \
R
ruri 已提交
7 8 9 10 11 12 13 14 15
       --lr_strategy=cosine_decay \
       --lr=0.1 \
       --num_epochs=200 \
       --model_save_dir=output/ \
       --l2_decay=1e-4 \
       --use_mixup=True \
       --resize_short_size=256 \
       --use_label_smoothing=True \
       --label_smoothing_epsilon=0.1 \