فهرست منبع

remove image_shape from quant_aware_train since no inference model is saved in qat process

will-jl944 4 سال پیش
والد
کامیت
3ef1df8af8
1فایلهای تغییر یافته به همراه1 افزوده شده و 2 حذف شده
  1. 1 2
      dygraph/tutorials/slim/quantize/image_classification/mobilenetv2_qat.py

+ 1 - 2
dygraph/tutorials/slim/quantize/image_classification/mobilenetv2_qat.py

@@ -40,5 +40,4 @@ model.quant_aware_train(
     eval_dataset=eval_dataset,
     learning_rate=0.000025,
     save_dir='output/mobilenet_v2/quant',
-    use_vdl=True,
-    infer_image_shape=[224, 224])
+    use_vdl=True)