Skip to content

Commit 3716ff4

Browse files
authored
update coco training default params
1 parent a2890cf commit 3716ff4

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

readme.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -147,7 +147,7 @@ Check out this [tutorial](tutorial/train_shape.ipynb) if you are new to this. Yo
147147
obj_list: ['person', 'bicycle', 'car', ...]
148148

149149

150-
## 3.a. Train on coco from scratch
150+
## 3.a. Train on coco from scratch(not necessary)
151151

152152
# train efficientdet-d0 on coco from scratch
153153
# with batchsize 12
@@ -159,7 +159,7 @@ Check out this [tutorial](tutorial/train_shape.ipynb) if you are new to this. Yo
159159
# The first few epoches will be rather unstable,
160160
# it's quite normal when you train from scratch.
161161

162-
python train.py -c 0 --batch_size 12
162+
python train.py -c 0 --batch_size 64 --optim sgd --lr 8e-2
163163

164164
## 3.b. Train a custom dataset from scratch
165165

0 commit comments

Comments
 (0)