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

Co-authored-by: Tom Benson <[email protected]>
LLNL · Feb 27, 2023 · bc412e5 · bc412e5
1 parent de1e304
commit bc412e5
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
@@ -19,7 +19,7 @@ def make_data_reader(num_classes=1000, data_path=None):
     if data_path is not None:
         print("Setting up data reader")
         train_data_dir = os.path.join(data_path, 'train')
-        test_data_dir = os.path.join(data_path, "val")
+        test_data_dir = os.path.join(data_path, 'val')
         #FIXME: This might not match the directory structure of the user's
         # dataset. We could infer the labels from the directory structure,
         # but the data reader is expecting a file.