浏览代码

update ppyolov2 tutorial training configuration

will-jl944 4 年之前
父节点
当前提交
5e9aacb875
共有 1 个文件被更改,包括 3 次插入2 次删除
  1. 3 2
      dygraph/tutorials/train/object_detection/ppyolov2.py

+ 3 - 2
dygraph/tutorials/train/object_detection/ppyolov2.py

@@ -50,13 +50,14 @@ model = pdx.models.PPYOLOv2(
 # API说明:https://github.com/PaddlePaddle/PaddleX/blob/release/2.0-rc/paddlex/cv/models/detector.py#L155
 # 各参数介绍与调整说明:https://paddlex.readthedocs.io/zh_CN/develop/appendix/parameters.html
 model.train(
-    num_epochs=365,
+    num_epochs=170,
     train_dataset=train_dataset,
     train_batch_size=8,
     eval_dataset=eval_dataset,
+    pretrain_weights='COCO',
     learning_rate=0.005 / 12,
     warmup_steps=1000,
     warmup_start_lr=0.0,
-    lr_decay_epochs=[243],
+    lr_decay_epochs=[105, 135, 150],
     save_interval_epochs=5,
     save_dir='output/ppyolov2_r50vd_dcn')