- Ubuntu 16.04
- pytorch=0.4.1 or pytorch=0.1
- Anaconda3, python3.6 or python3.5
- Install pytorch and other requirements by
pip install requirements.txt. cd ROOT_PATH. Make sure execute following cmd in the root directory of project.- train.py
pip install requirements.txt.cd ROOT_PATH. Make sure execute following cmd in the root directory of project.