SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

fix run_clip README (#18332)

Browse Source 
Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2022-08-02 19:14:46 +02:00
committed by GitHub
parent 2959d09072
commit 5546fb61ab
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
examples/pytorch/contrastive-image-text/README.md
View File

@@ -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: Having downloaded COCO dataset manually you should be able to load with the `ydshieh/coc_dataset_script` dataset loading script:
```py ```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) 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 \ python examples/pytorch/contrastive-image-text/run_clip.py \
--output_dir ./clip-roberta-finetuned \ --output_dir ./clip-roberta-finetuned \
--model_name_or_path ./clip-roberta \ --model_name_or_path ./clip-roberta \
--data_dir ./data \ --data_dir $PWD/data \
--dataset_name ydshieh/coco_dataset_script \ --dataset_name ydshieh/coco_dataset_script \
--dataset_config_name=2017 \ --dataset_config_name=2017 \
--image_column image_path \ --image_column image_path \