Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
5546fb61
Unverified
Commit
5546fb61
authored
Aug 02, 2022
by
Yih-Dar
Committed by
GitHub
Aug 02, 2022
Browse files
fix run_clip README (#18332)
Co-authored-by:
ydshieh
<
ydshieh@users.noreply.github.com
>
parent
2959d090
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
5 additions
and
2 deletions
+5
-2
examples/pytorch/contrastive-image-text/README.md
examples/pytorch/contrastive-image-text/README.md
+5
-2
No files found.
examples/pytorch/contrastive-image-text/README.md
View file @
5546fb61
...
...
@@ -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
\
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment