Преглед на файлове

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