Update README.md
Browse files
README.md
CHANGED
|
@@ -46,6 +46,8 @@ We tested the performance of the model on 1000 held-out test data and observed a
|
|
| 46 |
|
| 47 |
## Usage
|
| 48 |
|
|
|
|
|
|
|
| 49 |
```python
|
| 50 |
# torch packages
|
| 51 |
import torch
|
|
@@ -80,6 +82,9 @@ if __name__ == "__main__":
|
|
| 80 |
print(model)
|
| 81 |
|
| 82 |
```
|
|
|
|
|
|
|
|
|
|
| 83 |
|
| 84 |
## Resources
|
| 85 |
|
|
|
|
| 46 |
|
| 47 |
## Usage
|
| 48 |
|
| 49 |
+
Make sure to clone the repo and use the following code snippet to load the transformer model
|
| 50 |
+
|
| 51 |
```python
|
| 52 |
# torch packages
|
| 53 |
import torch
|
|
|
|
| 82 |
print(model)
|
| 83 |
|
| 84 |
```
|
| 85 |
+
### Source code
|
| 86 |
+
|
| 87 |
+
Source code used to train the model is linked in this [github](https://github.com/m-np/pytorch-transformer)
|
| 88 |
|
| 89 |
## Resources
|
| 90 |
|