ZixuanNi / examples

Deep Learning Examples

Deep Learning Examples

A repository of deep learning examples.

Getting Started

• Pick an example from the list below
• Use your local GPU machine, or get the GPUs you need in Lambda Cloud
• Follow the instructions in the examples README.md

Examples

• Finetune Stable Diffusion text-to-image model
• YoloV5 - object detection example using YoloV5
• GPTNeoX - Large Language Model Multi-Node Distributed Training
• Experiment Tracking
• PyTorch DDP - Multi node training with PyTorch DDP, torch.distributed.launch, torchrun and mpirun