Neural Turing Machine is a neural network (LSTM Cell) coupled with an external memory with which the neural network interacts (read and write) using attention processes. NTMs can be used as a model-based technique for meta-learning.
- PyTorch
- Torchvision
- Numpy
- Tqdm
- Run the train.py script for training the model with default settings. For options,
python train.py -h - Run the Predict.ipynb notebook for making predictions and visualizing the results.
- Alternatively, run this repository directly in colab
- Neural Turing Machines https://arxiv.org/abs/1410.5401
- Code is based on https://github.com/vlgiitr/ntm-pytorch
- https://www.niklasschmidinger.com/posts/2019-12-25-neural-turing-machines/
- Implementing Neural Turing Machines https://arxiv.org/abs/1807.08518