From 8e7aba3b5da27690b4216d16b87c2d5928f18709 Mon Sep 17 00:00:00 2001 From: Baek JeongHun Date: Tue, 25 Feb 2020 20:27:31 +0900 Subject: [PATCH] Update README.md --- README.md | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index 4b76650..2d50f4d 100644 --- a/README.md +++ b/README.md @@ -22,7 +22,8 @@ The difference between our paper and ICDAR challenge is summarized [here](https: ## Getting Started ### Dependency -- This work was tested with PyTorch 1.3.1, CUDA 9.0, python 3.6 and Ubuntu 16.04.
You may need `pip3 install torch==1.3.1`. In the paper, expriments were performed with **PyTorch 0.4.1**. +- This work was tested with PyTorch 1.3.1, CUDA 10.1, python 3.6 and Ubuntu 16.04.
You may need `pip3 install torch==1.3.1`.
+In the paper, expriments were performed with **PyTorch 0.4.1, CUDA 9.0**. - requirements : lmdb, pillow, torchvision, nltk, natsort ``` pip3 install lmdb pillow torchvision nltk natsort