Skip to content

Latest commit

 

History

History
executable file
·
60 lines (47 loc) · 3.48 KB

README.md

File metadata and controls

executable file
·
60 lines (47 loc) · 3.48 KB

Deep learning models in PyTorch

NOTE: This project is not actively maintained anymore. Feel free to send a PR or fork it.

This repository contains deep learning models built in PyTorch. Intended for learning PyTorch, this repo is made understandable for someone with basic python and deep learning knowledge. All kinds of important deep learning models are implemented here. Links to the papers are also given.

The structure for each project is consistent:

  • model.py - Contains the model of the neural network.
  • train.py - Preprocessing the data and training the network.
  • test.py - Infering from the trained network.
  • any other supporting files.

Common folders:

  • Trained folder - Trained models can will be saved here.
  • Datasets folder - Datasets downloaded will be stored here.

Best practices like PEP8, dropout, batch normalization, suitable optimizers are used.

IMP: For training on the CPU remove '.cuda()' wherever you find it.

Models

1. CNNs

2. RNNs

3. GANs

4. Reinforcement

5. Others

References: