diff --git a/use_case_examples/cifar/cifar_brevitas_training/evaluate_torch_cml.py b/use_case_examples/cifar/cifar_brevitas_training/evaluate_torch_cml.py index ff3dff4cb..ac7ee54bd 100644 --- a/use_case_examples/cifar/cifar_brevitas_training/evaluate_torch_cml.py +++ b/use_case_examples/cifar/cifar_brevitas_training/evaluate_torch_cml.py @@ -139,7 +139,7 @@ def main(args): "--rounding_threshold_bits", nargs="+", type=int, - default=[None], + default=[6], help="Number of bits to target with rounding.", ) parser.add_argument(