Update implement.py
Browse files- 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 |
-
|
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 = "/
|
36 |
text_embedding = 768
|
37 |
-
text_tokenizer = "/
|
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
|