Skip to content

GitLab

Unverified Commit 8e0d1b95 authored by Vasilis Vryniotis's avatar Vasilis Vryniotis Committed by GitHub
Browse files

Fix incorrect recipe for SSDlite320 (#6536)

parent f56e6f63
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
references/detection/README.md
View file @ 8e0d1b95
......@@ -68,7 +68,7 @@ torchrun --nproc_per_node=8 train.py\
torchrun --nproc_per_node=8 train.py\
--dataset coco --model ssdlite320_mobilenet_v3_large --epochs 660\
--aspect-ratio-group-factor 3 --lr-scheduler cosineannealinglr --lr 0.15 --batch-size 24\
--weight-decay 0.00004 --data-augmentation ssdlite --weights-backbone MobileNet_V3_Large_Weights.IMAGENET1K_V1
--weight-decay 0.00004 --data-augmentation ssdlite
```
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment