Minor Review: Exercise PyTorch Simple Neural Network