This will be used for learning deeplearning using Pytorch