Update README.md
Browse files
README.md
CHANGED
|
@@ -1,6 +1,7 @@
|
|
| 1 |
---
|
| 2 |
license: apache-2.0
|
| 3 |
---
|
|
|
|
| 4 |
class TransformerBlock(nn.Module):
|
| 5 |
def __init__(self, sizeVector=256, numHeads=8, dropout=0.1):
|
| 6 |
super().__init__()
|
|
|
|
| 1 |
---
|
| 2 |
license: apache-2.0
|
| 3 |
---
|
| 4 |
+
```python
|
| 5 |
class TransformerBlock(nn.Module):
|
| 6 |
def __init__(self, sizeVector=256, numHeads=8, dropout=0.1):
|
| 7 |
super().__init__()
|