WorkFlow extracts basic process of training a neural network, and handles the logging and the visualization of multiple loss variables. It allows you to focus on the parts that most deserve your valuable efforts, such as network structure and optimization strategy.
Install Pytorch
-
Install virtualenv: https://virtualenv.pypa.io/en/stable
-
Build a new virtualenv environment (this will create a folder in your current directory).
$ virtualenv pytorch
- Go to the new folder and activate the environment(https://virtualenv.pypa.io/en/stable/userguide/#usage)
$ source bin/activate
- Clone pytorch and build: https://github.com/pytorch/pytorch
$ git clone --recursive https://github.com/pytorch/pytorch
$ cd pytorch
$ python setup.py install
- Install dependencies
$ pip install numpy pyyaml mkl mkl-include setuptools cmake cffi typing
Install torchvision and visdom
$ pip install torchvision
$ pip install visdom
Run the visdom server
$ python -m visdom.server
Open another terminal
$ python sample.py