diff --git a/README.md b/README.md index a94e60c..ae937b5 100644 --- a/README.md +++ b/README.md @@ -23,7 +23,10 @@ $ make split ``` ### Requirements -Python (3.6.3), pytorch (0.4.1), torchvision (0.2.1), skimage (0.14.1), numpy (1.15.2), Jupyter Notebook(4.4.0) +Code is written in Python 3.6.3. [Here](requirements.txt) are all requirements, hit the following to install: +```bash +pip3 install -r requirements.txt +``` ### Network training