Skip to content

This repo contains the source code accompanying a scientific paper with the same name.

License

Notifications You must be signed in to change notification settings

sunhongmin225/meta-learning-lstm

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

50 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

meta-learning-lstm

This repo contains the code for the following paper: https://openreview.net/pdf?id=rJY0-Kcll

Dependencies

The following libaries are necessary:

Training

Splits corresponding to meta-training, meta-validation, and meta-testing are placed in data/miniImagenet/. Download corresponding imagenet images and place in folder called images and place folder in data/miniImagenet/.

To train a model:

th train/run-train.lua --task [1-shot or 5-shot task] --data config.imagenet --model [model name]

For example, to run matching-nets:

th train/run-train.lua --task config.5-shot-5-class --data config.imagenet --model config.baselines.train-matching-net

And, to run LSTM meta-learner for 5-shot task:

th train/run-train.lua --task config.5-shot-5-class --data config.imagenet --model config.lstm.train-imagenet-5shot

Contact

For questions about miniImagenet format, please contact Sachin Ravi at email given in the paper.

About

This repo contains the source code accompanying a scientific paper with the same name.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Lua 90.8%
  • Python 9.2%