Update the requirements for PyTorch.

sczhou · Jul 29, 2019 · 3d9eb12 · 3d9eb12
1 parent d7b4c70
commit 3d9eb12
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 21 deletions.
diff --git a/README.md b/README.md
@@ -16,7 +16,7 @@ You could download the pretrained model (21.5MB) of STFANet from [[Here]](https:
 
 - Linux (tested on Ubuntu 14.04/16.04)
 - Python 2.7+
-- Pytorch 0.4.1
+- Pytorch 0.4.1.post2
 - easydict
 - tensorboardX
 

diff --git a/models/FAC/README.md~ b/models/FAC/README.md~
diff --git a/requirements.txt b/requirements.txt
@@ -4,7 +4,8 @@ json
 numpy
 matplotlib
 scipy
+torch==0.4.1.post2
+torchvision
 opencv-python
 pprint
-torchvision
-tensorboardX
+tensorboardX