Skip to content

kaishengtai/torch-ntm

master
Switch branches/tags

Name already in use

A tag already exists with the provided branch name. Many Git commands accept both tag and branch names, so creating this branch may cause unexpected behavior. Are you sure you want to create this branch?
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 
 
 
 
 
 
 
 
 
 
 

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. NTM models with multiple read/write heads are supported.

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