https://coursevania.com/courses/introduction-to-pytorch-crash-course