ACAI (1807.07543) Pytorch Implementation based on kylemcdonald's gist
Python3.6 + Pytorch 1.0
Setup:
virtualenv -p /usr/bin/python3.6 ./py36+pytorch1
source ./py36+pytorch1/bin/activate
pip install https://download.pytorch.org/whl/cu80/torch-1.0.0-cp36-cp36m-linux_x86_64.whl
pip install -r requirement.txt
To visualize and remote visualize :
tensorboard --logdir=./log
ssh -L LOCAL_PORT:IP:REMOTE_PORT USER_NAME@SERVER_IP