will-jl944 4 år sedan
förälder
incheckning
107688bb8e
1 ändrade filer med 2 tillägg och 2 borttagningar
  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')