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

Update model.py

This commit is contained in:
Canberk Baykal 2021-06-29 16:13:53 +03:00
parent cfcffb90e6
commit d038cfaf59
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
clip/model.py
View File

@ -365,7 +365,7 @@ class CLIP(nn.Module):
logits_per_text = logit_scale * text_features @ image_features.t()
# shape = [global_batch_size, global_batch_size]
return logits_per_image, logits_per_text
return logits_per_image, logits_per_text, logit_scale
def convert_weights(model: nn.Module):