myownskyW7
commited on
Commit
•
a52d70f
1
Parent(s):
f8e6ab8
Update build_mlp.py
Browse files- build_mlp.py +2 -2
build_mlp.py
CHANGED
@@ -77,8 +77,8 @@ class CLIPVisionTower(nn.Module):
|
|
77 |
embedding_size = pos_embed_checkpoint.shape[-1]
|
78 |
num_extra_tokens = 1
|
79 |
new_num = new_size**2 + num_extra_tokens
|
80 |
-
print('Position interpolate from %dx%d to %dx%d' %
|
81 |
-
|
82 |
extra_tokens = pos_embed_checkpoint[:, :num_extra_tokens]
|
83 |
# only the position tokens are interpolated
|
84 |
pos_tokens = pos_embed_checkpoint[:, num_extra_tokens:]
|
|
|
77 |
embedding_size = pos_embed_checkpoint.shape[-1]
|
78 |
num_extra_tokens = 1
|
79 |
new_num = new_size**2 + num_extra_tokens
|
80 |
+
#print('Position interpolate from %dx%d to %dx%d' %
|
81 |
+
# (orig_size, orig_size, new_size, new_size))
|
82 |
extra_tokens = pos_embed_checkpoint[:, :num_extra_tokens]
|
83 |
# only the position tokens are interpolated
|
84 |
pos_tokens = pos_embed_checkpoint[:, num_extra_tokens:]
|