BOIL: Towards Representation Change for Few-shot Learning
This repository is the official implementation of "BOIL: Towards Representation Change for Few-shot Learning" on ICLR2021. Our implementations are relied on Torchmeta.
Requirements
We run our code in the following environment using Anaconda.
- Python >= 3.5
- Pytorch == 1.4
- torchvision == 0.5
If you use Pytorch version above 1.5 (which is the latest version at this moment) and torchvision above 0.6, you may encounter problem. In that case, you are encouraged to change to the version in our environment.
To install requirements:
pip install -r requirements.txt
Training
If you want to train 4conv network in the paper, run this command:
./run_4conv.sh
If you want to train ResNet-12 in the paper, run this command:
./run_resnet.sh
If you want to see and change the arguments of training code, run this command:
python3 main.py --help
Evaluation
To evaluate the model(s) and see the results, please refer to the analysis.ipynb