Training CNN with Mixed Precision in PyTorch 1.6
Simple Example on MNIST dataset illustrating Mixed Precision Training.
deep-learning pytorch code tutorial

  • Simple Example on MNIST dataset illustrating Mixed Precision Training.
  • Demonstrates the new inbuilt Mixed Precision Training.
  • Shows tricks like Gradient Penalty, and combining this procedure with normal training.
  • Shows how to train EfficientNet with these tricks.

Don't forget to tag @oke-aditya in your comment, otherwise they may not be notified.

Authors original post
Deep Learning | Computer Vision
Share this project
Similar projects
TabNet : Attentive Interpretable Tabular Learning
PyTorch implementation of TabNet paper.
BLiTZ — A Bayesian Neural Network library for PyTorch
Bayesian Layers in Torch Zoo is a simple and extensible library to create Bayesian Neural Network layers on the top of PyTorch.
Computer Vision Pretrained Models
A collection of computer vision pre-trained models.
Torchlayers
Shape inference for PyTorch (like Keras) & new layers