Skip to content

BoyuanJiang/matching-networks-pytorch

Repository files navigation

This is the pytorch implement of Matching Networks for One Shot Learning architecture

Train

If you want to train the model,simply run the code

python mainOmniglot.py 

You can set fce = True if you want use Full Context Embeddings.You can use the follow set in mainOmniglot.py for a 5-way one shot learning.

classes_per_set = 5
samples_per_class = 1

Result

After about 30 epoches,you can achieve about 97% accuracy on train set and 96% on val and test set. result

Acknowledgement

This work refer to gitabcworld's implement,detail athttps://github.com/AntreasAntoniou/MatchingNetworks.

About

Matching Networks for one shot learning

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages