SeemG commited on
Commit
e9078d1
·
verified ·
1 Parent(s): be42ff0

Update clip_model.py

Browse files
Files changed (1) hide show
  1. 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 = "/content/flickr30k_images/flickr30k_images"
22
- captions_path = "/content"
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