/pytorch-cifar

Neural Network Models on CIFAR with PyTorch

Primary LanguageJupyter Notebook

Deep Learning Models with PyTorch on CIFAR

Implementation of neural network models on CIFAR with Pytorch.

Dependencies

  • PyTorch 1.0+

Models

  • LeNet
  • MLP
  • VGG
  • ResNet
  • ResNet V2 (with pre-activation)
  • DenseNet