Skip to content

New Features, PyTorch 1.7.1 PyTorch Lightning 1.1 Support

Compare
Choose a tag to compare
@oke-aditya oke-aditya released this 13 Dec 06:49
· 15 commits to master since this release
ea2c74b

This new release cleans up our codebases internally. This makes the library more robust and easier to maintain.
Also, we have new examples and prototype APIs.

Code Cleanup and Bug Fixes

  • We cleaned up code internally and created a stable requirements.txt file. Now, we pin versions in our requirements instead of hardcoding them in CI checks.
  • It is recommended to install PyTorch as mentioned on PyTorch website and NOT run pip install -r requirements.txt.
  • Our requirements.txt file helps us to view Dependabot alerts as well as see sub dependencies graph.

#56 Cleaning up Type Annotations
#60 Fixed some typos and small fixes.
#63 Drop timm requirements.
#65 Clean up of codebase like Pl_bolts, making it easier to create a model zoo
#77 Fixing CI and file encoding issue over MacOS and windows.
#69 Enhanced Dummy Detection dataset for better internal testing
#79 Refactored tests to check for normalized bounding boxes and class starts.
#88 Added requirements.txt file which makes it easier to keep consistency.
#90 Fixed Detr Name and weights.

New Models and Tutorials

#76 Supports Wide Resnet for Classification as well as FPN based backbone in Detection.
#81 Example to train models with Weights and Biases.
#74 Example to train DETR over COCO Dataset.
#68 SWAV Weights for Resnet models in Detection and Classification tasks.

Prototype APIs

As a small release, we have started prototyping new APIs. We expect them to be stable in subsequent releases.

#86 Losses API for common losses in Computer Vision tasks.
#80 Layers API for layers pertaining to Computer Vision.

Documentation

  • We are planning to build Sphinx Documentation with autodoc features. Thanks to @SauravMaheshkar .
  • This is work in progress and contributions are welcome

#54 Adds Sphinx Docs
#67 Fixes a Few issues with Sphinx
#58 Added CI to deploy Docs over GitHub pages.

Since this release is fully compatible with PyTorch Lightning 1.1, people can use all features of Lightning such as Shared Training, etc.

Super thanks to @hassiahk @zhiqwang @zlapp @SauravMaheshkar for making this release possible!