ibrim commited on
Commit
eb8df1e
·
verified ·
1 Parent(s): 88147e5

Update implement.py

Browse files
Files changed (1) hide show
  1. implement.py +3 -4
implement.py CHANGED
@@ -13,8 +13,7 @@ import torch.nn.functional as F
13
  import timm
14
  from transformers import DistilBertModel, DistilBertConfig, DistilBertTokenizer
15
  import os
16
- os.environ['HTTPS_PROXY']="http://185.46.212.90:80/"
17
- os.environ['HTTP_PROXY']="http://185.46.212.90:80/"
18
  class CFG:
19
  debug = False
20
  image_path = "/raid/users/mohammadibrahim-st/TSAI/OpenAI-CLIP/Flicker-8k/Images"
@@ -32,9 +31,9 @@ class CFG:
32
 
33
  model_name = 'resnet50'
34
  image_embedding = 2048
35
- text_encoder_model = "/raid/users/mohammadibrahim-st/Models/BertDistil"
36
  text_embedding = 768
37
- text_tokenizer = "/raid/users/mohammadibrahim-st/Models/BertDistil"
38
  max_length = 200
39
 
40
  pretrained = True # for both image encoder and text encoder
 
13
  import timm
14
  from transformers import DistilBertModel, DistilBertConfig, DistilBertTokenizer
15
  import os
16
+
 
17
  class CFG:
18
  debug = False
19
  image_path = "/raid/users/mohammadibrahim-st/TSAI/OpenAI-CLIP/Flicker-8k/Images"
 
31
 
32
  model_name = 'resnet50'
33
  image_embedding = 2048
34
+ text_encoder_model = "distilbert/distilbert-base-uncased"
35
  text_embedding = 768
36
+ text_tokenizer = "distilbert/distilbert-base-uncased"
37
  max_length = 200
38
 
39
  pretrained = True # for both image encoder and text encoder