localhorst/CLIP_on_Tesla_K20Xm
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge branch 'main' into feature/make_installable

Browse Source
This commit is contained in:
Jong Wook Kim
2021-01-30 01:35:13 +09:00
parent 9c29c9cc8d 578a1d3e2e
commit a19dcf967b
3 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
clip/model.py
View File

@ -329,11 +329,11 @@ class CLIP(nn.Module):
# cosine similarity as logits
logit_scale = self.logit_scale.exp()
logits_per_iamge = logit_scale * image_features @ text_features.t()
logits_per_image = logit_scale * image_features @ text_features.t()
logits_per_text = logit_scale * text_features @ image_features.t()
# shape = [global_batch_size, global_batch_size]
return logits_per_iamge, logits_per_text
return logits_per_image, logits_per_text
def convert_weights(model: nn.Module):