Skip to content

Lightning Support, Better visualizations

Latest
Latest
Compare
Choose a tag to compare
@oke-aditya oke-aditya released this 19 Oct 08:58
· 2 commits to master since this release
v0.3.0
ec60c87
This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
GPG key ID: 4AEE18F83AFDEB23
Expired
Learn about vigilant mode.

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.

Assets 2
Loading