Commit
•
ff974d2
1
Parent(s):
81ac6a9
initial commit
Browse files
README.md
CHANGED
|
@@ -33,4 +33,6 @@ print(embedding_mean.shape) # expect to be 768
|
|
| 33 |
# embedding with max pooling
|
| 34 |
embedding_max = torch.max(hidden_states[0], dim=0)[0]
|
| 35 |
print(embedding_max.shape) # expect to be 768
|
| 36 |
-
```
|
|
|
|
|
|
|
|
|
| 33 |
# embedding with max pooling
|
| 34 |
embedding_max = torch.max(hidden_states[0], dim=0)[0]
|
| 35 |
print(embedding_max.shape) # expect to be 768
|
| 36 |
+
```
|
| 37 |
+
license: mit
|
| 38 |
+
---
|