Neural Network With Pytorch Results Graph