This is an experimental code base
The biggest difference between MAML and pre-training weights:Pre-training weights minimize only for original task loss. MAML can minimize all task loss with a few steps of training.
- Pull repository.
git clone https://github.com/verages/MAML.git
- You need to install some dependency package.
cd MAML
pip installl -r requirements.txt
- Download the Omiglot dataset and maml weights.
wget https://github.com/verages/MAML/releases/download/v0.1/Omniglot.tar
wget https://github.com/verages/MAML/releases/download/v0.1/maml.h5
tar -xvf Omniglot.tar
- Run evaluate.py, you'll see the difference between MAML and random initialization weights.
python evaluate.py
- You should set same parameters in config.py.
n_way = "number of classes"
k_shot = "number of support set"
q_query = "number of query set"
- Start training.
python train.py
- Running tensorboard to monitor the training process.
tensorboard --logdir=./summary