Update applications/vision/data/imagenet/__init__.py

Co-authored-by: Tom Benson <[email protected]>
LLNL · Feb 27, 2023 · 30dd65e · 30dd65e
1 parent bc412e5
commit 30dd65e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/applications/vision/data/imagenet/__init__.py b/applications/vision/data/imagenet/__init__.py
@@ -47,7 +47,7 @@ def make_data_reader(num_classes=1000, data_path=None):
             test_data_dir = imagenet_dir(data_set='val')
             test_label_file = imagenet_labels(data_set='val')
     else:
-        raise RuntimeError(f'ImageNet data paths are unknown for current compute center ({compute_center}). Set the "data-path" argument to the location of your dataset.')
+        raise RuntimeError(f'ImageNet data paths are unknown for current compute center ({compute_center}). Set "--data-path" to the location of your dataset.')
 
     # Check that data paths are accessible
     if not os.path.isdir(train_data_dir):