Skip to content

Commit 01e6309

Browse files
committed
add paper link
1 parent 81be580 commit 01e6309

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

README.md

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
## Transformer PyTorch implementation
2-
This repository contains **Transformer** implementation used to **translate Korean sentence into English sentence**.
2+
This repository contains [**Transformer**](https://papers.nips.cc/paper/7181-attention-is-all-you-need.pdf) implementation used to **translate Korean sentence into English sentence**.
33

44
I used translation dataset for NMT, but you can apply this model to any sequence to sequence (i.e. text generation) tasks such as text summarization, response generation, ..., etc.
55

@@ -29,7 +29,7 @@ Example:
2929

3030
### Requirements
3131

32-
- Following libraries are fundamental to this repo.
32+
- Following libraries are fundamental to this repository.
3333
- You should install PyTorch via official [Installation guide](https://pytorch.org/get-started/locally/).
3434
- To use spaCy model which is used to tokenize english sentence, download English model by running `python -m spacy download en_core_web_sm`.
3535

@@ -58,7 +58,6 @@ test the model and predict user's input sentence.
5858
python build_pickle.py --kor_vocab KOREAN_VOCAB_SIZE --eng_vocab ENGLISH_VOCAB_SIZE
5959
```
6060

61-
6261
- For training, run `main.py` with train mode (which is default option)
6362

6463
```
@@ -91,7 +90,7 @@ eng> I am going to meet my girlfriend tomorrow
9190
```
9291

9392
<p align="center">
94-
<img width="400" height="600" src="img/result2.png">
93+
<img width="350" height="600" src="img/result2.png">
9594
</p>
9695

9796
```

0 commit comments

Comments
 (0)