Allowed a default value for the model argument

This commit is contained in:
Kinar 2023-11-13 21:44:27 -08:00
parent f8add5ad42
commit 252cca72e7

View File

@ -46,7 +46,7 @@ def main():
parser = argparse.ArgumentParser(
formatter_class=argparse.ArgumentDefaultsHelpFormatter)
parser.add_argument(
'--model', help='Path to image classification model.', required=True,
'--model', help='Path to image classification model.', required=False,
default='classifier.tflite')
parser.add_argument(
'--iterations', help='Number of iterations for benchmarking.', type=int,