소스 검색

Update mobilenetv2_quant.py

Jason 5 년 전
부모
커밋
841f5ed41f
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      tutorials/slim/quant/image_classification/mobilenetv2_quant.py

+ 1 - 1
tutorials/slim/quant/image_classification/mobilenetv2_quant.py

@@ -28,4 +28,4 @@ dataset = pdx.datasets.ImageNet(
                 transforms=model.test_transforms)
 
 # 开始量化
-pdx.slim.export_quant_model(model, dataset, save_dir='./quant_mobilenet', cache_dir='./tmp')
+pdx.slim.export_quant_model(model, dataset, batch_size=4, batch_num=10, save_dir='./quant_mobilenet', cache_dir='./tmp')