Views
No views yet
1python3 train_ssd.py \
2 --pretrained-ssd=models/mb2-ssd-lite-mp-0_686.pth \
3 --net=mb2-ssd-lite \
4 --dataset-type=voc \
5 --data=data/voc \
6 --model-dir=models/output \
7 --num-workers=8 \
8 --batch-size=16 \
9 --num-epochs=56 \
10 --t-0=8 \
11 --validation-epochs=1 \
12 --lr=0.0001 \
13 --extra-layers-lr=0.001 \
14 --base-net-lr=0.000001 \
15 --weight-decay=0.000001 \
16 --validation-mean-ap \
17 --scheduler=cosine-warmrestart \
18 --no-augment