fix typo

Xilinx · May 1, 2024 · c0f596b · c0f596b
1 parent f5d5e8e
commit c0f596b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/brevitas_examples/imagenet_classification/ptq/ptq_evaluate.py b/src/brevitas_examples/imagenet_classification/ptq/ptq_evaluate.py
@@ -350,7 +350,7 @@ def main():
     model = model.to(dtype)
 
     # Generate calibration and validation dataloaders
-    if args.repostory == 'torchvision':
+    if args.repository == 'torchvision':
         resize_shape = model_config['resize_shape']
         center_crop_shape = model_config['center_crop_shape']
         inception_preprocessing = model_config['inception_preprocessing']