Skip to content

Commit 1737744

Browse files
author
krammnic
committed
fix docs
1 parent f637b24 commit 1737744

File tree

1 file changed

+5
-2
lines changed
  • torchtune/datasets/multimodal

1 file changed

+5
-2
lines changed

torchtune/datasets/multimodal/_vqa.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -51,8 +51,11 @@ def vqa_dataset(
5151
in the filepath in ``data_files``, and set ``split="train"``. See `Hugging Face's
5252
<https://huggingface.co/docs/datasets/en/package_reference/loading_methods#datasets.load_dataset.path>`_
5353
``load_dataset`` for more details.
54-
image_dir (str): path to the directory containing the images as you are expected to download the COCO dataset
55-
before using. Default is "coco/".
54+
image_dir (str): path to the directory containing the images that is prepended to all image
55+
paths in the dataset. For example, if ``image_dir="/home/user/dataset/"` and the sample image path
56+
was ``"images/1.jpg"``, the final image path that will be loaded is ``"/home/user/dataset/images/1.jpg"``.
57+
If None, assume images are available in current working directory or are located
58+
on a remote url. For text-only, leave as None. Default is None.
5659
column_map (Optional[Dict[str, str]]): a mapping to change the expected "input",
5760
"output", and "image" column names to the actual column names in the dataset. Keys should be "input",
5861
"output", and "image, and values should be the actual column names.

0 commit comments

Comments
 (0)