Skip to content

yusugomori/deeplearning-pytorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

459bdf0 · Mar 22, 2019

History

36 Commits
Mar 7, 2019
Mar 22, 2019
Mar 7, 2019
Mar 22, 2019

Repository files navigation

Deep Learning with PyTorch 1.X

Implementations of neural network models with torch (>=1.0)

See also implementations with TesorFlow 2.0 here.

Requirements

  • PyTorch >= 1.0
$ pip install torch torchvision

Models

  • Logistic Regression
  • MLP
  • LeNet
  • ResNet (ResNet34, ResNet50)
  • Encoder-Decoder (LSTM)
  • EncoderDecoder (Attention)
  • Transformer
  • Deep Q-Network
models/
├── dqn_cartpole.py
├── encoder_decoder_attention.py
├── encoder_decoder_lstm.py
├── lenet_mnist.py
├── logistic_regression_mnist.py
├── mlp_mnist.py
├── resnet34_fashion_mnist.py
├── resnet50_fashion_mnist.py
├── transformer.py
│
└── layers/
    ├── Attention.py
    ├── DotProductAttention.py
    ├── Flatten.py
    ├── GlobalAvgPool2d.py
    ├── MultiHeadAttention.py
    ├── PositionalEncoding.py
    └── ScaledDotProductAttention.py

About

Deep learning model zoo with PyTorch 1.X

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages