This repository contains PyTorch code examples. Below are the list of example projects.
Project Name | Description |
---|---|
NN_CIFAR10 |
A simple neural network model for CIFAR-10 dataset. |
VGG-16 |
VGG-16 CNN architecture. |
PyTorch_DataLoader |
PyTorch's DataLoader class. To load data more efficiently |
Multi_GPU |
PyTorch code to utilize more than 1 GPU during training. |
Transfer_Learning |
Transfer Learning by freezing convolutional layers weights. |
MIT