Explorar o código

change ppyolo tutorial training configuration

will-jl944 %!s(int64=4) %!d(string=hai) anos
pai
achega
1d695a0540
Modificáronse 1 ficheiros con 3 adicións e 2 borrados
  1. 3 2
      dygraph/tutorials/train/object_detection/ppyolo.py

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

@@ -46,12 +46,13 @@ model = pdx.models.PPYOLO(num_classes=num_classes, backbone='ResNet50_vd_dcn')
 # API说明:https://github.com/PaddlePaddle/PaddleX/blob/release/2.0-rc/paddlex/cv/models/detector.py#L154
 # API说明:https://github.com/PaddlePaddle/PaddleX/blob/release/2.0-rc/paddlex/cv/models/detector.py#L154
 # 各参数介绍与调整说明:https://paddlex.readthedocs.io/zh_CN/develop/appendix/parameters.html
 # 各参数介绍与调整说明:https://paddlex.readthedocs.io/zh_CN/develop/appendix/parameters.html
 model.train(
 model.train(
-    num_epochs=405,
+    num_epochs=450,
     train_dataset=train_dataset,
     train_dataset=train_dataset,
     train_batch_size=8,
     train_batch_size=8,
     eval_dataset=eval_dataset,
     eval_dataset=eval_dataset,
+    pretrain_weights='COCO',
     learning_rate=0.005 / 12,
     learning_rate=0.005 / 12,
-    warmup_steps=1000,
+    warmup_steps=500,
     warmup_start_lr=0.0,
     warmup_start_lr=0.0,
     save_interval_epochs=5,
     save_interval_epochs=5,
     lr_decay_epochs=[243, 324],
     lr_decay_epochs=[243, 324],