Skip to content
/ ntm Public

A PyTorch Implementation of Neural Turing Machine

Notifications You must be signed in to change notification settings

nerdimite/ntm

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Neural Turing Machine

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.

Requirements

  • PyTorch
  • Torchvision
  • Numpy
  • Tqdm

Usage

  1. Run the train.py script for training the model with default settings. For options, python train.py -h
  2. Run the Predict.ipynb notebook for making predictions and visualizing the results.
  3. Alternatively, run this repository directly in colab Open In Colab

Credits