ECS_189G_Project Stage 3: Main branch is MINIST Look at other branches for ORL and CIFAR Stage 4 Please look at stage_4_classification and stage_4_generation branches for classification and generation tasks, respectively.