Learning to Compare: Relation Network for Few-Shot Learning
Pytorch Implementation for Paper: Learning to Compare: Relation Network for Few-Shot Learning download mini-imagenet and make it looks like: mini-imagenet/ ├── images ├── n0210891500001298.jpg ├── n0287152500001298.jpg … ├── test.csv ├── val.csv └── train.csv LearningToCompare-Pytorch/ ├── compare.py ├── MiniImagenet.py ├── Readme.md ├── repnet.py ├── train.py └── utils.py python train.py current code support multi-gpus on single machine training, to disable it and train on single machine,just set device_ids=[0] and downsize batch size according to your gpu memory capacity.make sure ckpt directory exists, […]
Read more