Update README.md
Browse files
README.md
CHANGED
@@ -31,4 +31,20 @@ tokenizer = Tokenizer(model.config.input_encoding_map, model.config.output_encod
|
|
31 |
output = model(tokenizer('1 1 0 0 1 0').unsqueeze(0))
|
32 |
print(tokenizer.decode(output.argmax(2)))
|
33 |
>>> [['bos', '1', '1', '1', '1', '1', '1']]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
34 |
```
|
|
|
31 |
output = model(tokenizer('1 1 0 0 1 0').unsqueeze(0))
|
32 |
print(tokenizer.decode(output.argmax(2)))
|
33 |
>>> [['bos', '1', '1', '1', '1', '1', '1']]
|
34 |
+
```
|
35 |
+
|
36 |
+
|
37 |
+
### 🙏Cite🙏
|
38 |
+
|
39 |
+
|
40 |
+
###### If you are interested in our paper, please feel free to cite it.
|
41 |
+
```
|
42 |
+
@misc{weng2023neural,
|
43 |
+
title={Neural Comprehension: Language Models with Compiled Neural Networks},
|
44 |
+
author={Yixuan Weng and Minjun Zhu and Fei Xia and Bin Li and Shizhu He and Kang Liu and Jun Zhao},
|
45 |
+
year={2023},
|
46 |
+
eprint={2304.01665},
|
47 |
+
archivePrefix={arXiv},
|
48 |
+
primaryClass={cs.CL}
|
49 |
+
}
|
50 |
```
|