parent
beba48f353
commit
43c953e231
|
@ -287,7 +287,7 @@ class CLIP(nn.Module):
|
|||
self.ln_final = LayerNorm(transformer_width)
|
||||
|
||||
self.text_projection = nn.Parameter(torch.empty(transformer_width, embed_dim))
|
||||
self.logit_scale = nn.Parameter(torch.ones([]))
|
||||
self.logit_scale = nn.Parameter(torch.FloatTensor([np.log(1/0.07)]))
|
||||
|
||||
self.initialize_parameters()
|
||||
|
||||
|
|
Loading…
Reference in New Issue