Skip to content

0.2.0

Compare
Choose a tag to compare
@creafz creafz released this 24 Dec 07:44
· 5 commits to master since this release
68de6fc

Minor changes

  • AutoAlbument now uses the instantiate function from Hydra to create instances of Classification and Semantic Segmentation models. This change allows using custom models inside AutoAlbument. See the CIFAR10 example that uses a custom ResNet-18 model.
  • AutoAlbument now requires timm>=0.3.2 and segmentation-models-pytorch>=0.1.3.
  • All arguments to SearchDataset classes in examples are now defined in Hydra configs.
  • Fixed typos in code.