We used Torch for our experiments. Here we provide a PyTorch version which is not well-tested.

Dependencies

pip install -r requirements.txt

Training

python train.py --restore=0

Set restore=1 to resume training from a checkpoint.

Pre-trained model

We provide the pretrained checkpoint here.

Testing

python train.py --task=test