https://drive.google.com/drive/folders/1Vuslm6C7XjNTnv1MD_OnhyVm07FTmJeQ?usp=sharing
Main branch is MINIST
Look at other branches for ORL and CIFAR
Please look at stage_4_classification and stage_4_generation branches for classification and generation tasks, respectively.