소스 검색

qat ready

will-jl944 4 년 전
부모
커밋
107688bb8e
1개의 변경된 파일2개의 추가작업 그리고 2개의 파일을 삭제
  1. 2 2
      dygraph/tutorials/slim/quantize/object_detection/yolov3_qat.py

+ 2 - 2
dygraph/tutorials/slim/quantize/object_detection/yolov3_qat.py

@@ -48,8 +48,8 @@ model.quant_aware_train(
     train_batch_size=8,
     eval_dataset=eval_dataset,
     learning_rate=0.0001 / 8,
-    warmup_steps=1000,
+    warmup_steps=100,
     warmup_start_lr=0.0,
-    save_interval_epochs=5,
+    save_interval_epochs=1,
     lr_decay_epochs=[30, 45],
     save_dir='output/yolov3_darknet53/quant')