瀏覽代碼

reset lr for ppyolo tutorial

FlyingQianMM 5 年之前
父節點
當前提交
cfbac17fbc
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      tutorials/train/object_detection/ppyolo.py

+ 1 - 1
tutorials/train/object_detection/ppyolo.py

@@ -52,7 +52,7 @@ model.train(
     train_dataset=train_dataset,
     train_batch_size=8,
     eval_dataset=eval_dataset,
-    learning_rate=0.0005,
+    learning_rate=0.000125,
     lr_decay_epochs=[210, 240],
     save_dir='output/ppyolo',
     use_vdl=True)