AI Workshop: Build a Neural Network with PyTorch Lightning

Introduction

• AI workshop: Build a neural network with PyTorch Lightning
• Prerequisites
• Quick overview of neural networks

1. Building a Neural Network with PyTorch

• Setting up the virtual environment
• Loading and exploring regression data
• Preprocessing data for training
• Creating a simple neural network
• Setting up the dataset and DataLoader
• Training a neural network using PyTorch
• Visualizing losses and evaluating models
• Building and training a more complex neural network

2. Using PyTorch Lightning to Build a Regression Model

• Encapsulating data using a LightningDataModule
• Encapsulating a model using a LightningModule
• Training the model using the PyTorch Lightning Trainer

3. Using PyTorch Lightning to Build a Classification Model

• Loading and exploring classification data
• Creating a LightningDataModule
• Creating a LightningModule
• Training a classification model and evaluating metrics

Conclusion

• Summary and next steps