Pytorch-cnn-trainer

Glad to announce the release 0.3.0 of PyTorch CNN Trainer.

Features

- PyTorch Lightning trainer is now avaliable. It supports all torchvision models as of now. (26)
- [BC BREAKING] Dictionary object with metrics is returned from fit instead of True 23
- Better plotting utility to visualize results quickly (25)
- Utility to turn off NVIDIA Profilers. This might results in faster training (21)

Examples
- New examples for lightning based trainer (26)

Tests and CI
- Tests are added for lightning (26)
- Dropped Black code formatting.

Contributors

This release was possible due to contributions from premalrupnur, oke-aditya . Glad to have more people here.

PyTorch CNN Trainer

A complete CNN trainer package that can train all CNNs in torchvision and timm models.
It can fine-tune as well as train from scratch.

Features

- [x] Support PyTorch image models (timm) training and transfer learning.
- [x] Quantization Aware training example.
- [x] Early stopping with patience.
- [x] Support torchvision models trainging and transfer learning.
- [x] Support torchvision quantized models transfer learning.
- [x] Support for Mixed Precision Training.
- [x] L2 Norm Gradient Penalty.
- [x] SWA Stochastic weighted Averaging support for training.
- [x] Add Keras Like fit method.
- [x] Sanity Check method.