STM_net in PyTorch

Prerequisites

git clone https://github.com/ZeHeru/STM_net.git
cd STM_net

Install PyTorch and 0.4+ and other dependencies (e.g., torchvision, visdom and dominate).
- For pip users, please type the command pip install -r requirements.txt.
- For Conda users, you can create a new Conda environment using conda env create -f environment.yml.

Name		Name	Last commit message	Last commit date
Latest commit History 13 Commits
data		data
models		models
options		options
util		util
README.md		README.md
requirements.txt		requirements.txt
test.py		test.py
train.py		train.py