Skip to content

Commit bd94e75

Browse files
authored
Fix CLIP logit scale inplace clip behavior (#130)
* add swin transformer * add swin transformer * add swin transformer * fix clip logit scale behavior
1 parent cbef352 commit bd94e75

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

passl/modeling/backbones/clip.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -314,7 +314,8 @@ def encode_text(self, text):
314314
return x
315315

316316
def clip_logit_scale(self):
317-
self.logit_scale.clip(-4.6, 4.6)
317+
logit_scale_buffer = self.logit_scale.clip_(-4.6, 4.6)
318+
logit_scale_buffer._share_buffer_to(self.logit_scale)
318319

319320
def forward(self, image, text, is_train=True):
320321
image_features = self.encode_image(image)

0 commit comments

Comments
 (0)