fujenchu / matchingNet

implementation of Matching Net in Pytorch

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