hongduantian / dense-sparse-dense-reptile

Dense-Sparse-Dense Reptile

This repository contains the code and experiments for the paper:
Meta Learning with Network Pruning (ECCV '20)

Requirements

tensorflow-gpu=1.4
python3.6
tqdm

Getting the data

miniimagenet dataset

tieredimagenet dataset

Please download the compressed tar files from: https://github.com/renmengye/few-shot-ssl-public

mkdir -p ../tieredImagenet/data  
tar -xvf tiered-imagenet.tar  
mv *.pkl ../tieredImagenet/data  

Run experiments

For convenience, we provided 4 demos as seen in the repository. Directly run the .sh file or you can change some parameters in the file to run some other experiments.

sh ./[filename]