Build a Neural Network using Pytorch and Google collab