Fix names

Xilinx · Oct 14, 2024 · 28b10cc · 28b10cc
1 parent 8fba690
commit 28b10cc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/brevitas_examples/imagenet_classification/ptq/ptq_evaluate.py b/src/brevitas_examples/imagenet_classification/ptq/ptq_evaluate.py
@@ -47,7 +47,7 @@ def parse_type(v, default_type):
         return default_type(v)
 
 
-def validate(args):
+def validate_args(args):
     if args.learned_round:
         assert args.target_backend == "layerwise", "Currently, learned round is only supported with target-backend=layerwise"
 
@@ -285,7 +285,7 @@ def generate_ref_input(args, device, dtype):
 
 def main():
     args = parser.parse_args()
-    validate(args)
+    validate_args(args)
     dtype = getattr(torch, args.dtype)
 
     random.seed(SEED)