"docs/source/vscode:/vscode.git/clone" did not exist on "7fc80724dace3e27fd9d540228fa4d95fbf94970"
Unverified Commit 5546fb61 authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

fix run_clip README (#18332)


Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent 2959d090
......@@ -43,7 +43,10 @@ cd ..
Having downloaded COCO dataset manually you should be able to load with the `ydshieh/coc_dataset_script` dataset loading script:
```py
COCO_DIR = "data"
import os
import datasets
COCO_DIR = os.path.join(os.getcwd(), "data")
ds = datasets.load_dataset("ydshieh/coco_dataset_script", "2017", data_dir=COCO_DIR)
```
......@@ -84,7 +87,7 @@ Finally, we can run the example script to train the model:
python examples/pytorch/contrastive-image-text/run_clip.py \
--output_dir ./clip-roberta-finetuned \
--model_name_or_path ./clip-roberta \
--data_dir ./data \
--data_dir $PWD/data \
--dataset_name ydshieh/coco_dataset_script \
--dataset_config_name=2017 \
--image_column image_path \
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment