New comer to deep learning and pytorch