Imported from https://github.com/PatriciaRodrigues1994/pytorch-facebook-udacity-challenge