Lab4: Pytorch, and Artificial Neural Networks (contd)