update readme, docker for tensorboardX issue

changqi97 · Sep 4, 2018 · 8385c60 · 8385c60
1 parent 86698ef
commit 8385c60
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -66,5 +66,5 @@ RUN pip3 install http://download.pytorch.org/whl/cu90/torch-0.4.0-cp35-cp35m-lin
 RUN pip3 install torchvision 
 
 ## other packages
-RUN pip3 install tensorboardX
+RUN pip3 install tensorboardX==1.2
 RUN pip3 install tensorflow
diff --git a/README.md b/README.md
@@ -32,7 +32,10 @@ Please cite our paper if you find the code or dataset useful for your research.
 ### Prerequisites
 - Python 3.5 or Python 3.6
 - Pytorch 0.4.0 and torchvision (https://pytorch.org/)
-- [TensorboardX](https://github.com/lanpa/tensorboard-pytorch)
+- [TensorboardX](https://github.com/lanpa/tensorboard-pytorch) ***1.2***
+```
+pip3 install tensorboardX==1.2
+```
 - [Tensorflow](https://www.tensorflow.org/) (for tensorboard usage)
 - We provide a Docker file for building the environment based on CUDA 9.0, CuDNN 7.1, and Ubuntu 16.04.