Pytorch version

hkchengrex · Aug 19, 2020 · fdc6dbe · fdc6dbe
1 parent 2bfb4ac
commit fdc6dbe
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -18,6 +18,8 @@ Here are some refinement results on high-resolution images.
 
 ## Quick Start
 
+Tested on PyTorch 1.0 -- though higher versions would likely work for inference as well.
+
 Check out [this folder](https://github.com/hkchengrex/CascadePSP/tree/master/segmentation-refinement). We have built a pip package that can refine an input image with two lines of code.
 
 Install with

diff --git a/docs/installation.md b/docs/installation.md
@@ -2,7 +2,7 @@
 
 ## Dependencies
 
-CascadePSP is tested on PyTorch 1.0 though higher version would likely work as well. 
+CascadePSP is tested on PyTorch 1.0 though higher version would likely work for inference as well.
 For installing PyTorch, please refer to the [Pytorch website](https://pytorch.org/).  
 
 We recommend using the anaconda distribution which should contain most of the required dependencies. Other dependencies can be installed by: