소스 검색

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)