Imported from https://github.com/VivekUdupa/CPSC8810-DeepLearning_vkoodli_shonnah