Skip to content

A Neural Turing Machine implementation in Torch.

Notifications You must be signed in to change notification settings

AjayTalati/torch-ntm

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

A Neural Turing Machine in Torch

A Torch implementation of the Neural Turing Machine model described in this paper by Alex Graves, Greg Wayne and Ivo Danihelka.

This implementation uses an LSTM controller with one read head and one write head.

Requirements

Torch7 (of course), as well as the following libraries:

penlight

nn

optim

nngraph

All the above dependencies can be installed using luarocks. For example:

luarocks install nngraph

Usage

For the copy task:

th tasks/copy.lua

For the associative recall task:

th tasks/recall.lua

About

A Neural Turing Machine implementation in Torch.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Lua 100.0%