This repository is for reproducing a toy-experiment result in figure 2 of Discrete Key-Value Bottleneck.
You can train and check the result by:
$ python train.py
The result:
You can check each image at images folder.
DISCLAIMER: THIS IS NOT AN OFFICIAL IMPLEMENTATION