diff --git a/README.md b/README.md index 950aa1a..38ce592 100644 --- a/README.md +++ b/README.md @@ -1,2 +1,15 @@ -# pytorch-pl-hydra-templates -Different template codes for Deep Learning with PyTorch. +# Template codes for Deep learning with Pytorch +This repo provides different pytorch implementation for training a deep learning model. It uses a simple classification task example for [CIFAR10 Dataset](https://www.cs.toronto.edu/~kriz/cifar.html) to show: + 1. A simple vanilla, [single-GPU implementation](#single-gpu-pytorch-implementation) + 2. A [multi-GPU, distributed data parallel implementation](multi-gpu-pytorch-implementation) + 3. A [Pytorch-ligtning implementation](pytorch-ligtning-implementation) along with tracking and visualization in TensorBoard + 4. A [Pytorch-ligtning Hydra implementation](pytorch-ligtning-hydra-implementation) for rapid experimentation and prototyping using new models/datasets + +# Single-GPU implementation + +# Multi-GPU distributed data parallel implementation + +# Pytorch-lightning implementation + +# Pytorch-lightning Hydra implementation +