cosmetic change to quantization shell script
[jacinto-ai/pytorch-jacinto-ai-devkit.git] / run_quantization.sh
1 # Quantization
3 ## =====================================================================================
4 ## Quantization Aware Training
5 ## =====================================================================================
6 #
7 #### Image Classification - Quantization Aware Training - MobileNetV2
8 #python ./scripts/train_classification_main.py --dataset_name image_folder_classification --model_name mobilenetv2_tv_x1 --data_path ./data/datasets/image_folder_classification \
9 #--pretrained https://download.pytorch.org/models/mobilenet_v2-b0353104.pth \
10 #--batch_size 64 --quantize True --epochs 25 --epoch_size 0.1 --lr 1e-5 --evaluate_start False
11 #
12 #
13 #### Image Classification - Quantization Aware Training - MobileNetV2(Shicai) - a TOUGH MobileNetV2 pretrained model
14 #python ./scripts/train_classification_main.py --dataset_name image_folder_classification --model_name mobilenetv2_shicai_x1 --data_path ./data/datasets/image_folder_classification \
15 #--pretrained ./data/modelzoo/pytorch/image_classification/imagenet1k/shicai/mobilenetv2_shicai_rgb.pth \
16 #--batch_size 64 --quantize True --epochs 25 --epoch_size 0.1 --lr 1e-5 --evaluate_start False
17 #
18 #
19 #### Semantic Segmentation - Quantization Aware Training for MobileNetV2+DeeplabV3Lite
20 #python ./scripts/train_segmentation_main.py --dataset_name cityscapes_segmentation --model_name deeplabv3lite_mobilenetv2_tv --data_path ./data/datasets/cityscapes/data --img_resize 384 768 --output_size 1024 2048 --gpus 0 1 \
21 #--pretrained ./data/modelzoo/pytorch/semantic_seg/cityscapes/jacinto_ai/deeplabv3lite_mobilenetv2_tv_768x384_best.pth \
22 #--batch_size 6 --quantize True --epochs 150 --lr 1e-5 --evaluate_start False
23 #
24 #
25 #### Semantic Segmentation - Quantization Aware Training for MobileNetV2+UNetLite
26 #python ./scripts/train_segmentation_main.py --dataset_name cityscapes_segmentation --model_name unetlite_pixel2pixel_aspp_mobilenetv2_tv --data_path ./data/datasets/cityscapes/data --img_resize 384 768 --output_size 1024 2048 --gpus 0 1 \
27 #--pretrained ./data/modelzoo/pytorch/semantic_seg/cityscapes/jacinto_ai/unet_aspp_mobilenetv2_tv_768x384_best.pth \
28 #--batch_size 6 --quantize True --epochs 150 --lr 1e-5 --evaluate_start False
32 ## =====================================================================================
33 ## Post Training Calibration & Quantization - this is fast, but may not always yield best quantized accuracy (not recommended)
34 ## =====================================================================================
35 #
36 #### Image Classification - Post Training Calibration & Quantization - ResNet50
37 #python ./scripts/train_classification_main.py --phase calibration --dataset_name image_folder_classification --model_name resnet50_x1 --data_path ./data/datasets/image_folder_classification \
38 #--pretrained https://download.pytorch.org/models/resnet50-19c8e357.pth \
39 #--batch_size 64 --quantize True --epochs 1 --epoch_size 0.1 --evaluate_start False
40 #
41 #
42 #### Image Classification - Post Training Calibration & Quantization - MobileNetV2
43 #python ./scripts/train_classification_main.py --phase calibration --dataset_name image_folder_classification --model_name mobilenetv2_tv_x1 --data_path ./data/datasets/image_folder_classification \
44 #--pretrained https://download.pytorch.org/models/mobilenet_v2-b0353104.pth \
45 #--batch_size 64 --quantize True --epochs 1 --epoch_size 0.1 --evaluate_start False
46 #
47 #
48 #### Image Classification - Post Training Calibration & Quantization for a TOUGH MobileNetV2 pretrained model
49 #python ./scripts/train_classification_main.py --phase calibration --dataset_name image_folder_classification --model_name mobilenetv2_shicai_x1 --data_path ./data/datasets/image_folder_classification \
50 #--pretrained ./data/modelzoo/pytorch/image_classification/imagenet1k/shicai/mobilenetv2_shicai_rgb.pth \
51 #--batch_size 64 --quantize True --epochs 1 --epoch_size 0.1 --evaluate_start False
52 #
53 #
54 ### Semantic Segmentation - Post Training Calibration &  Quantization for MobileNetV2+DeeplabV3Lite
55 #python ./scripts/train_segmentation_main.py --phase calibration --dataset_name cityscapes_segmentation --model_name deeplabv3lite_mobilenetv2_tv --data_path ./data/datasets/cityscapes/data --img_resize 384 768 --output_size 1024 2048 --gpus 0 1 \
56 #--pretrained ./data/modelzoo/pytorch/semantic_seg/cityscapes/jacinto_ai/deeplabv3lite_mobilenetv2_tv_768x384_best.pth \
57 #--batch_size 6 --quantize True --epochs 1 --evaluate_start False
58 #
59 #
60 ### Semantic Segmentation - Post Training Calibration &  Quantization for MobileNetV2+UNetLite
61 #python ./scripts/train_segmentation_main.py --phase calibration --dataset_name cityscapes_segmentation --model_name unetlite_pixel2pixel_aspp_mobilenetv2_tv --data_path ./data/datasets/cityscapes/data --img_resize 384 768 --output_size 1024 2048 --gpus 0 1 \
62 #--pretrained ./data/modelzoo/pytorch/semantic_seg/cityscapes/jacinto_ai/unet_aspp_mobilenetv2_tv_768x384_best.pth \
63 #--batch_size 6 --quantize True --epochs 1 --evaluate_start False
67 ## =====================================================================================
68 ## Acuracy Evaluation with Post Training Quantization - cannot save quantized model - only accuracy evaluation
69 ## =====================================================================================
71 #### Image Classification - Accuracy Estimation with Post Training Quantization - MobileNetV2
72 #python ./scripts/train_classification_main.py --phase validation --dataset_name image_folder_classification --model_name mobilenetv2_tv_x1 --data_path ./data/datasets/image_folder_classification \
73 #--pretrained https://download.pytorch.org/models/mobilenet_v2-b0353104.pth \
74 #--batch_size 64 --quantize True
76 #### Image Classification - Accuracy Estimation with Post Training Quantization - ResNet50
77 #python ./scripts/train_classification_main.py --phase validation --dataset_name image_folder_classification --model_name resnet50_x1 --data_path ./data/datasets/image_folder_classification \
78 #--pretrained https://download.pytorch.org/models/resnet50-19c8e357.pth \
79 #--batch_size 64 --quantize True
81 #### Image Classification - Accuracy Estimation with Post Training Quantization - A TOUGH MobileNetV2 pretrained model
82 #python ./scripts/train_classification_main.py --phase validation --dataset_name image_folder_classification --model_name mobilenetv2_shicai_x1 --data_path ./data/datasets/image_folder_classification \
83 #--pretrained ./data/modelzoo/pytorch/image_classification/imagenet1k/shicai/mobilenetv2_shicai_rgb.pth \
84 #--batch_size 64 --quantize True
86 #### Semantic Segmentation - Accuracy Estimation with Post Training Quantization - MobileNetV2+DeeplabV3Lite
87 #python ./scripts/train_segmentation_main.py --phase validation --dataset_name cityscapes_segmentation --model_name deeplabv3lite_mobilenetv2_tv --data_path ./data/datasets/cityscapes/data --img_resize 384 768 --output_size 1024 2048 --gpus 0 1 \
88 #--pretrained './data/modelzoo/pytorch/semantic_seg/cityscapes/jacinto_ai/deeplabv3lite_mobilenetv2_tv_768x384_best.pth' \
89 #--batch_size 1 --quantize True
91 #### Semantic Segmentation - Accuracy Estimation with Post Training Quantization - MobileNetV2+UNetLite
92 #python ./scripts/train_segmentation_main.py --phase validation --dataset_name cityscapes_segmentation --model_name unetlite_pixel2pixel_aspp_mobilenetv2_tv --data_path ./data/datasets/cityscapes/data --img_resize 384 768 --output_size 1024 2048 --gpus 0 1 \
93 #--pretrained ./data/modelzoo/pytorch/semantic_seg/cityscapes/jacinto_ai/unet_aspp_mobilenetv2_tv_768x384_best.pth \
94 #--batch_size 1 --quantize True