modified annotations env variable

GATEOverflow · Oct 17, 2024 · 372c633 · 372c633
1 parent a08e402
commit 372c633
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/script/app-mlperf-inference-nvidia/customize.py b/script/app-mlperf-inference-nvidia/customize.py
@@ -141,7 +141,7 @@ def preprocess(i):
         dataset_path = env['CM_DATASET_OPENIMAGES_PATH']
         #return {'return': 1, 'error': 'error'}
 
-        annotations_path = env['CM_DATASET_OPENIMAGES_VALIDATION_ANNOTATIONS_FILE_PATH']
+        annotations_path = env['CM_DATASET_OPENIMAGES_ANNOTATIONS_DIR_PATH']
         target_data_path_dir = os.path.join(env['MLPERF_SCRATCH_PATH'], 'data', 'open-images-v6-mlperf')
         if not os.path.exists(target_data_path_dir):
             cmds.append(f"mkdir -p {target_data_path_dir}")