Skip to content

Commit

Permalink
Update Readme.md
Browse files Browse the repository at this point in the history
  • Loading branch information
garg-aayush authored May 4, 2022
1 parent 7e3a79c commit 65d2fc2
Showing 1 changed file with 15 additions and 2 deletions.
17 changes: 15 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
@@ -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

0 comments on commit 65d2fc2

Please sign in to comment.