lightning-flash
lightning-flash copied to clipboard
Your PyTorch AI Factory - Flash enables you to easily configure and run complex AI recipes for over 15 tasks across 7 data domains
## 🚀 Feature I'd like to be able to train iterable-style datasets instead of just map-style datasets. (a map-style dataset in PyTorch has `__getitem__` and `__len__`, whereas iterable-style datasets only...
## 🚀 Feature Add support for self supervised learning, especially for multimodal input and improve the current vissl backed features. ### Motivation I have been working on multimodal self supervised...
## 🐛 Bug Hi! I need to train the slow_50 model for spatio temporal detection. I didn't find a VideoClassificationData for loading AVA dataset, so, I decided to build a...
## 🚀 Feature Please consider porting more heads and backbones for object detection task. ### Motivation The number of heads and backbones for the task of object detection remains rather...
### Discussed in https://github.com/PyTorchLightning/lightning-flash/discussions/1288 Originally posted by **sirtris** April 11, 2022 Hi, I have pre-trained my own network backbone using the `ImageEmbedder`. I would now use this model in a...
## What does this PR do? Fixes #1452 Currently, I was able to only find the changes in `flash/core/utilities/imports.py` ## Before submitting - [x] Was this **discussed/approved** via a Github...
## 🚀 Feature We have released https://github.com/Lightning-AI/utilities/ which contains multiple utilities that are shared across the codebase. ### Motivation Avoid protected imports Avoid duplicated code ### Pitch Go through the...
## What does this PR do? Part of https://github.com/Lightning-AI/lightning/issues/12545. Same as https://github.com/Lightning-AI/lightning/pull/14469. ## Before submitting - [x] Was this **discussed/approved** via a Github issue? (no need for typos and docs...
## 🚀 Feature Add `weight_decay_filter` and `lars_adaptation_filter` to LARS ### Motivation weight decay typically shouldn't be applied to BatchNorm. See [fast.ai](https://www.fast.ai/2018/08/10/fastai-diu-imagenet/) and this [pytorch discuss thread](https://discuss.pytorch.org/t/weight-decay-in-the-optimizers-is-a-bad-idea-especially-with-batchnorm/16994). The facebook [vicreg code](https://github.com/facebookresearch/vicreg/blob/main/main_vicreg.py#L245)...