Fix coco2014 sample ids path

mlcommons · Dec 25, 2024 · 0d61848 · 0d61848
1 parent a90475d
commit 0d61848
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/script/get-dataset-coco2014/customize.py b/script/get-dataset-coco2014/customize.py
@@ -24,7 +24,7 @@ def postprocess(i):
     env = i['env']
     if env.get('CM_GENERATE_SAMPLE_ID', '') == "yes":
         env['CM_COCO2014_SAMPLE_ID_PATH'] = os.path.join(
-            os.getcwd(), 'install', 'sample_ids.txt')
+            os.getcwd(), 'sample_ids.txt')
         print(env['CM_COCO2014_SAMPLE_ID_PATH'])
     if env.get('CM_DATASET_CALIBRATION', '') == "no":
         env['CM_DATASET_PATH_ROOT'] = os.getcwd()