Home Explore Help
Register Sign In
zhengchun
/
PaddleX
1
0
Fork 0
Files Issues 0 Pull Requests 0 Wiki
Browse Source

qat ready

will-jl944 4 years ago
parent
6c24d4fc34
commit
107688bb8e
1 changed files with 2 additions and 2 deletions
Split View Show Diff Stats
  1. 2 2
      dygraph/tutorials/slim/quantize/object_detection/yolov3_qat.py

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

@@ -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')