/matchingNet

implementation of Matching Net in Pytorch

Primary LanguagePythonApache License 2.0Apache-2.0

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