Technical report is published

kakaobrain · Apr 22, 2020 · 05503f0 · 05503f0
2 parents 34f344e + 2f8cbb4
commit 05503f0
Show file tree

Hide file tree

Showing 4 changed files with 19 additions and 18 deletions.
diff --git a/README.ko.md b/README.ko.md
@@ -177,11 +177,11 @@ torchgpipe 프로젝트는 [카카오브레인][]의 [이흥섭][], [정명룡][
 해당 라이브러리를 연구용으로 사용할 경우, 아래 BibTeX 링크를 인용해야 합니다.
 
 ```
-@misc{torchgpipe,
-  author       = {Lee, Heungsub and Jeong, Myungryong and Kim, Chiheon and
-                  Lim, Sungbin and Kim, Ildoo and Baek, Woonhyuk and Yoon, Boogeon},
-  title        = {torchgpipe, {A} {GPipe} implementation in {PyTorch}},
-  howpublished = {\url{https://github.com/kakaobrain/torchgpipe}},
-  year         = {2019}
+@article{kim2020torchgpipe,
+    title={torchgpipe: On-the-fly Pipeline Parallelism for Training Giant Models},
+    author={Chiheon Kim and Heungsub Lee and Myungryong Jeong and Woonhyuk Baek and Boogeon Yoon and Ildoo Kim and Sungbin Lim and Sungwoong Kim},
+    year={2020},
+    eprint={2004.09910},
+    archivePrefix={arXiv}
 }
 ```
diff --git a/README.md b/README.md
@@ -191,11 +191,11 @@ License 2.0](LICENSE).
 If you apply this library to any project and research, please cite our code:
 
 ```
-@misc{torchgpipe,
-  author       = {Lee, Heungsub and Jeong, Myungryong and Kim, Chiheon and
-                  Lim, Sungbin and Kim, Ildoo and Baek, Woonhyuk and Yoon, Boogeon},
-  title        = {torchgpipe, {A} {GPipe} implementation in {PyTorch}},
-  howpublished = {\url{https://github.com/kakaobrain/torchgpipe}},
-  year         = {2019}
+@article{kim2020torchgpipe,
+    title={torchgpipe: On-the-fly Pipeline Parallelism for Training Giant Models},
+    author={Chiheon Kim and Heungsub Lee and Myungryong Jeong and Woonhyuk Baek and Boogeon Yoon and Ildoo Kim and Sungbin Lim and Sungwoong Kim},
+    year={2020},
+    eprint={2004.09910},
+    archivePrefix={arXiv}
 }
 ```
diff --git a/docs/conf.py b/docs/conf.py
@@ -81,6 +81,7 @@
 
     'extra_nav_links': {
         'Source Code': 'https://github.com/kakaobrain/torchgpipe',
+        'Technical Report': 'https://arxiv.org/abs/2004.09910',
         'Original Paper': 'https://arxiv.org/abs/1811.06965',
     },
 }

diff --git a/docs/index.rst b/docs/index.rst
@@ -63,10 +63,10 @@ If you apply this library to any project and research, please cite our code:
 
 .. sourcecode:: bibtex
 
-   @misc{torchgpipe,
-     author       = {Lee, Heungsub and Jeong, Myungryong and Kim, Chiheon and
-                     Lim, Sungbin and Kim, Ildoo and Baek, Woonhyuk and Yoon, Boogeon},
-     title        = {torchgpipe, {A} {GPipe} implementation in {PyTorch}},
-     howpublished = {\url{https://github.com/kakaobrain/torchgpipe}},
-     year         = {2019}
+   @article{kim2020torchgpipe,
+       title={torchgpipe: On-the-fly Pipeline Parallelism for Training Giant Models},
+       author={Chiheon Kim and Heungsub Lee and Myungryong Jeong and Woonhyuk Baek and Boogeon Yoon and Ildoo Kim and Sungbin Lim and Sungwoong Kim},
+       year={2020},
+       eprint={2004.09910},
+       archivePrefix={arXiv}
    }