1. Please use imagenet_sampler.ipynb notebook to generate multiple episodes of mini-ImageNet and tiered-ImageNet. 
2. Next, use generate_data.sh to generate augmentation using the diffusion model. 