浏览代码

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)