Spaces:
Runtime error
Runtime error
Update clip_model.py
Browse files- clip_model.py +2 -2
clip_model.py
CHANGED
@@ -18,8 +18,8 @@ from transformers import DistilBertModel, DistilBertConfig, DistilBertTokenizer
|
|
18 |
|
19 |
class CFG:
|
20 |
debug = False
|
21 |
-
image_path = "/
|
22 |
-
captions_path = "
|
23 |
batch_size = 32
|
24 |
num_workers = 2
|
25 |
head_lr = 1e-3
|
|
|
18 |
|
19 |
class CFG:
|
20 |
debug = False
|
21 |
+
image_path = "/flickr30k_images/flickr30k_images"
|
22 |
+
captions_path = "."
|
23 |
batch_size = 32
|
24 |
num_workers = 2
|
25 |
head_lr = 1e-3
|