This is part 1 Face Classification for HW2P2. In addition to this, you should also finish part 2 Face verification

Given an image of a person’s face, the task of classifying the ID of the face is known as face classification. The input to your system will be a face image and you will have to predict the ID of the face. The ground truth will be present in the training data and the network will be doing an
Multi-class Classification to get the prediction. You are provided with a validation set for fine-tuning your model.