소스 검색

Merge pull request #301 from FlyingQianMM/develop_qh

reset lr for ppyolo tutorial
Jason 5 년 전
부모
커밋
5d76f9321d
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)