Skip to content

Commit 8e0d1b9

Browse files
authored
Fix incorrect recipe for SSDlite320 (#6536)
1 parent f56e6f6 commit 8e0d1b9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

references/detection/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -68,7 +68,7 @@ torchrun --nproc_per_node=8 train.py\
6868
torchrun --nproc_per_node=8 train.py\
6969
--dataset coco --model ssdlite320_mobilenet_v3_large --epochs 660\
7070
--aspect-ratio-group-factor 3 --lr-scheduler cosineannealinglr --lr 0.15 --batch-size 24\
71-
--weight-decay 0.00004 --data-augmentation ssdlite --weights-backbone MobileNet_V3_Large_Weights.IMAGENET1K_V1
71+
--weight-decay 0.00004 --data-augmentation ssdlite
7272
```
7373

7474

0 commit comments

Comments
 (0)