Skip to content

Commit cd26b1d

Browse files
authored
Update README.md
1 parent 70d782e commit cd26b1d

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

README.md

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -229,7 +229,7 @@ NOTE:
229229
#### EfficientDet-D0
230230

231231
Latest training run with .336 for D0 (on 4x 1080ti):
232-
`./distributed_train.sh 4 --model efficientdet_d0 -b 22 --amp --lr .12 --sync-bn --opt fusedmomentum --warmup-epochs 5 --lr-noise 0.4 0.9 --model-ema --model-ema-decay 0.9999`
232+
`./distributed_train.sh 4 /mscoco --model efficientdet_d0 -b 22 --amp --lr .12 --sync-bn --opt fusedmomentum --warmup-epochs 5 --lr-noise 0.4 0.9 --model-ema --model-ema-decay 0.9999`
233233

234234
These hparams above resulted in a good model, a few points:
235235
* the mAP peaked very early (epoch 200 of 300) and then appeared to overfit, so likely still room for improvement
@@ -238,6 +238,13 @@ These hparams above resulted in a good model, a few points:
238238
* drop_path (aka survival_prob / drop_connect) rate of 0.1, which is higher than the suggested 0.0 for D0 in official, but lower than the 0.2 for the other models
239239
* longer EMA period than default
240240

241+
#### EfficientDet-D1
242+
243+
Latest run with .394 mAP (on 4x 1080ti):
244+
`./distributed_train.sh 4 /mscoco --model efficientdet_d1 -b 10 --amp --lr .06 --sync-bn --opt fusedmomentum --warmup-epochs 5 --lr-noise 0.4 0.9 --model-ema --model-ema-decay 0.99995`
245+
246+
For this run I used some improved augmentations, still experimenting so not ready for release, should work well without them but will likely start overfitting a bit sooner and possibly end up a in the .385-.39 range.
247+
241248
VAL2017
242249
```
243250
Average Precision (AP) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.336251

0 commit comments

Comments
 (0)