瀏覽代碼

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