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
|