Aamir
commited on
Commit
•
6e5b002
1
Parent(s):
82f6eee
Update README.md
Browse files
README.md
CHANGED
@@ -2645,7 +2645,7 @@ model = SentenceTransformer(modules=[word_embedding_model, pooling_model])
|
|
2645 |
new_num_layers = 22 # 1D: set layer size
|
2646 |
model[0].auto_model.encoder.layer = model[0].auto_model.encoder.layer[:new_num_layers]
|
2647 |
|
2648 |
-
new_embedding_size = 768 #
|
2649 |
|
2650 |
# 3. encode
|
2651 |
embeddings = model.encode(
|
|
|
2645 |
new_num_layers = 22 # 1D: set layer size
|
2646 |
model[0].auto_model.encoder.layer = model[0].auto_model.encoder.layer[:new_num_layers]
|
2647 |
|
2648 |
+
new_embedding_size = 768 # 2D: set embedding size
|
2649 |
|
2650 |
# 3. encode
|
2651 |
embeddings = model.encode(
|