Renamed half to float16 for consistency

Signed-off-by: hkayann <[email protected]>
Xilinx · Feb 6, 2025 · ebe901b · ebe901b
1 parent 26412f8
commit ebe901b
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -2,7 +2,7 @@
 default_language_version:
     python: python3
 
-default_stages: [pre-commit]
+default_stages: [commit]
 
 exclude: '(src/brevitas/backport|docs)'
 

diff --git a/src/brevitas_examples/imagenet_classification/ptq/README.md b/src/brevitas_examples/imagenet_classification/ptq/README.md
@@ -71,7 +71,7 @@ usage: ptq_evaluate.py [-h] --calibration-dir CALIBRATION_DIR --validation-dir
                        [--batch-size-validation BATCH_SIZE_VALIDATION]
                        [--export-dir EXPORT_DIR] [--gpu GPU]
                        [--calibration-samples CALIBRATION_SAMPLES]
-                       [--model-name ARCH] [--dtype {float,bfloat16,half}]
+                       [--model-name ARCH] [--dtype {float,bfloat16,float16}]
                        [--target-backend {fx,layerwise,flexml}]
                        [--scale-factor-type {float_scale,po2_scale}]
                        [--act-bit-width ACT_BIT_WIDTH]
@@ -167,8 +167,8 @@ options:
                         vgg16 | vgg16_bn | vgg19 | vgg19_bn | vit_b_16 |
                         vit_b_32 | vit_h_14 | vit_l_16 | vit_l_32 |
                         wide_resnet101_2 | wide_resnet50_2 (default: resnet18)
-  --dtype {float,bfloat16,half}
-                        Data type to use (float for FP32, bfloat16, or half
+  --dtype {float,bfloat16,float16)
+                        Data type to use (float for FP32, bfloat16 for BF16, or float16
                         for FP16)
   --target-backend {fx,layerwise,flexml}
                         Backend to target for quantization (default: fx)

diff --git a/src/brevitas_examples/imagenet_classification/ptq/ptq_evaluate.py b/src/brevitas_examples/imagenet_classification/ptq/ptq_evaluate.py
@@ -89,8 +89,8 @@ def validate_args(args):
 parser.add_argument(
     '--dtype',
     default='float',
-    choices=['float', 'bfloat16', 'half'],
-    help='Data type to use (float for FP32, bfloat16, or half for FP16)')
+    choices=['float', 'bfloat16', 'float16'],
+    help='Data type to use (float for FP32, bfloat16 for BF16, or float16 for FP16)')
 parser.add_argument(
     '--target-backend',
     default='fx',