Bugfix in benchmark.py

cvg · Nov 21, 2023 · b1cd942 · b1cd942
1 parent 29f3e44
commit b1cd942
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/benchmark.py b/benchmark.py
@@ -189,7 +189,7 @@ def print_as_table(d, title, cnames):
             image0, image1 = [x.to(device) for x in inputs[pair_name]]
             runtimes = []
             for num_kpts in args.num_keypoints:
-                extractor.conf["max_num_keypoints"] = num_kpts
+                extractor.conf.max_num_keypoints = num_kpts
                 feats0 = extractor.extract(image0)
                 feats1 = extractor.extract(image1)
                 runtime = measure(
@@ -217,7 +217,7 @@ def print_as_table(d, title, cnames):
                 image0, image1 = [x.to(device) for x in inputs[pair_name]]
                 runtimes = []
                 for num_kpts in args.num_keypoints:
-                    extractor.conf["max_num_keypoints"] = num_kpts
+                    extractor.conf.max_num_keypoints = num_kpts
                     feats0 = extractor.extract(image0)
                     feats1 = extractor.extract(image1)
                     data = {