2.3 用PyTorch Lightning实现神经网络实例