Ver Fonte

update training configuration in ppyolo tutorial

will-jl944 há 4 anos atrás
pai
commit
c8a5dde909

+ 1 - 1
dygraph/requirements.txt

@@ -1,4 +1,4 @@
--r PaddleClas/requirements.txt
+-r ./PaddleClas/requirements.txt
 -r ./PaddleSeg/requirements.txt
 ./PaddleSeg
 -r ./PaddleDetection/requirements.txt

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

@@ -51,9 +51,9 @@ model.train(
     train_batch_size=8,
     eval_dataset=eval_dataset,
     learning_rate=0.005 / 12,
-    warmup_steps=1000,
+    warmup_steps=500,
     warmup_start_lr=0.0,
     save_interval_epochs=5,
-    lr_decay_epochs=[243, 324],
+    lr_decay_epochs=[160, 210, 230],
     save_dir='output/ppyolo_r50vd_dcn',
     use_vdl=True)