Quellcode durchsuchen

Update mobilenetv2_quant.py

Jason vor 5 Jahren
Ursprung
Commit
841f5ed41f
1 geänderte Dateien mit 1 neuen und 1 gelöschten Zeilen
  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')