This repo contains the code for Matching Networks for One Shot Learning in Pytorch
Please contact the authors for dataset (either miniImageNet
from Vinyals or new miniImageNet
from Ravi)
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
pretrained weights for miniImageNet on 5-way 1-shot is here
- add FCE from my Tensorflow implementation
- merge training and testing