Skip to content
This repository was archived by the owner on Oct 9, 2023. It is now read-only.

Adding TabNet and ObjectDetector

Choose a tag to compare

View all tags
@Borda Borda released this 12 Feb 19:33
· 919 commits to master since this release
0.2.0
0f6e202

[0.2.0] - 2021-02-12

Added

  • Added ObjectDetector Task (#56)
  • Added TabNet for tabular classification (#101)
  • Added support for more backbones(mobilnet, vgg, densenet, resnext) (#45)
  • Added backbones for image embedding model (#63)
  • Added SWAV and SimCLR models to imageclassifier + backbone reorg (#68)

Changed

  • Applied transform in FilePathDataset (#97)
  • Moved classification integration from vision root to folder (#86)

Fixed

  • Unfreeze default number of workers in datamodule (#57)
  • Fixed wrong label in FilePathDataset (#94)

Removed

  • Removed densenet161 duplicate in DENSENET_MODELS (#76)
  • Removed redundant num_features arg from Classification model (#88)
Assets 2
Loading