Explorar o código

update picodet tutorial

will-jl944 %!s(int64=4) %!d(string=hai) anos
pai
achega
70368c9480
Modificáronse 1 ficheiros con 7 adicións e 6 borrados
  1. 7 6
      tutorials/train/object_detection/picodet.py

+ 7 - 6
tutorials/train/object_detection/picodet.py

@@ -44,15 +44,16 @@ model = pdx.det.PicoDet(num_classes=num_classes, backbone='ESNet_l')
 # API说明:https://github.com/PaddlePaddle/PaddleX/blob/develop/docs/apis/models/detection.md
 # 各参数介绍与调整说明:https://github.com/PaddlePaddle/PaddleX/blob/develop/docs/parameters.md
 model.train(
-    num_epochs=300,
+    num_epochs=40,
     train_dataset=train_dataset,
-    train_batch_size=7,
+    train_batch_size=14,
     eval_dataset=eval_dataset,
     pretrain_weights='COCO',
     learning_rate=.05,
-    warmup_steps=300,
+    warmup_steps=30,
     warmup_start_lr=0.0,
-    save_interval_epochs=5,
-    lr_decay_epochs=[85, 135],
-    save_dir='output/piconet_esnet_l',
+    save_interval_epochs=2,
+    lr_decay_epochs=[8, 13],
+    use_ema=True,
+    save_dir='output/picodet_esnet_l',
     use_vdl=True)