Skip to content

Latest commit

 

History

History
25 lines (18 loc) · 734 Bytes

README.md

File metadata and controls

25 lines (18 loc) · 734 Bytes

matchingNet

This repo contains the code for Matching Networks for One Shot Learning in Pytorch


Usage

datasets

Please contact the authors for dataset (either miniImageNet from Vinyals or new miniImageNet from Ravi)

training

python visdom.server -port 5000
(or python -m visdom.server -port 5000)
python demo.py --gpuid 0 --logport 5000

feel free to modify config.py to change arguments

weights

pretrained weights for miniImageNet on 5-way 1-shot is here

todos

  • add FCE from my Tensorflow implementation
  • merge training and testing