pip install requirements.txt
- Download Omniglot dataset and extract the contents of images_background.zip and images_evaluation.zip to ./datasets/omniglot/
git clone [email protected]:brendenlake/omniglot.git
datasets/omniglot
|-- images_background
|-- images_evaluation
- Modify parameters in config.py, such as batchsize, lr ,update step, ...
- train model -> please set cfg.mode = True
- test model -> please set cfg.mode = False
- Run the main python script(trian or test)
python main.py