Optimize training params for tinierssd model (svhn dataset) (#155)

analogdevicesinc · Jun 16, 2022 · bfa4335 · bfa4335
1 parent 8eb1cf1
commit bfa4335
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/policies/qat_policy_svhn.yaml b/policies/qat_policy_svhn.yaml
@@ -1,4 +1,4 @@
 ---
-start_epoch: 15
+start_epoch: 25
 weight_bits: 8
 shift_quantile: 0.995
diff --git a/scripts/train_svhn_tinierssd.sh b/scripts/train_svhn_tinierssd.sh
@@ -1,2 +1,2 @@
 #!/bin/sh
-python train.py --deterministic --print-freq 200 --pr-curves --epochs 300 --optimizer Adam --lr 0.001 --wd 0 --model ai85tinierssd --use-bias --momentum 0.9 --weight-decay 5e-4 --dataset SVHN_74 --device MAX78000 --obj-detection --obj-detection-params parameters/obj_detection_params_svhn.yaml --batch-size 16 --qat-policy policies/qat_policy_svhn.yaml --validation-split 0 "$@"
+python train.py --deterministic --print-freq 200 --pr-curves --epochs 100 --optimizer Adam --lr 0.001 --wd 0 --model ai85tinierssd --use-bias --momentum 0.9 --weight-decay 5e-4 --dataset SVHN_74 --device MAX78000 --obj-detection --obj-detection-params parameters/obj_detection_params_svhn.yaml --batch-size 16 --qat-policy policies/qat_policy_svhn.yaml --validation-split 0 "$@"