floodsung / LearningToCompare_FSL

PyTorch code for CVPR 2018 paper: Learning to Compare: Relation Network for Few-Shot Learning (Few-Shot Learning part)
MIT License
1.04k stars 268 forks source link

Corrected depreciated functions and tested using ipython #35

Open HyoungsungKim opened 4 years ago

HyoungsungKim commented 4 years ago

As we know original version was implemented using python2 and pytorch 0.3.

So I corrected depreciated things to fix errors and warnings.

Furthermore, to understand how few-shot learning works, I tested using ipython.

I hope it is helpful to someone who just started to learn few-shot learning like me.

I implemented omniglot first and miniimagenet will be added soon.

Omniglot

If you liked it, please give a star :star: !