Create a Basic Neural Network Model - Deep Learning with PyTorch 5