From PyTorch to PyTorch Lightning — A gentle introduction

The Typical AI Research project

In a research project, we normally want to identify the following key components:

• the model(s)

• the data

• the loss

• the optimizer(s)