lightning-bolts
lightning-bolts copied to clipboard
Toolbox of models, callbacks, and datasets for AI/ML researchers.
## What does this PR do? Part of #839 - `pl_bolts.datamodules.sklearn_datamodule.SklearnDataModule` - `pl_bolts.datamodules.sklearn_datamodule.SklearnDataset` - `pl_bolts.datamodules.sklearn_datamodule.TensorDataset` ### Summary - Refactored `sklearn_datamodule.py`. (breaking change ❗ ) - Replaced `SlearnDataset` with `ArrayDataset`. (breaking...
Let's revision Bolts and breathe some fresh air into them! As outlined in #819 and on a Slack channel, we will revisit every single feature within Bolts. Please sign up...
## What does this PR do? Related to #839 - [x] pl_bolts.models.detection.yolo.yolo_config.py - [x] pl_bolts.models.detection.yolo.yolo_module.py - [x] pl_bolts.models.detection.yolo.yolo_layers.py Added some unit tests & yolo_giou config - [x] tests.models.yolo.test_yolo_config.py - [x]...
## What does this PR do? Part of https://github.com/Lightning-AI/lightning-bolts/issues/839 - [X] pl_bolts.callbacks.data_monitor.DataMonitorBase - [X] pl_bolts.callbacks.data_monitor.ModuleDataMonitor - [X] pl_bolts.callbacks.data_monitor.shape2str - [X] pl_bolts.callbacks.data_monitor.TrainingDataMonitor Minor changes only. Readability, small performance improvement, typehints Feedback...
## What does this PR do? Fixes part of #839 - `pl_bolts.models.self_supervised.simclr.transforms.GaussianBlur` (removed in favor of [torchvision's implementation](https://github.com/pytorch/vision/blob/96aa3d928c6faaf39defd846966421679244412d/torchvision/transforms/transforms.py#L1774-L1837)) - `pl_bolts.models.self_supervised.simclr.transforms.SimCLREvalDataTransform` (add unit test) - `pl_bolts.models.self_supervised.simclr.transforms.SimCLRFinetuneTransform` (add unit test and doc...
## 🚀 Feature Several new variants of the YOLO model have been proposed in the past couple of years. The current implementation in PyTorch Lightning Bolts is based on the...
## What does this PR do? The YOLO model has been largely refactored, so that it's easy to incorporate new features from different YOLO variants, such as different algorithms for...
## What does this PR do? - Part of https://github.com/Lightning-AI/lightning/pull/13831. - Changes the tag, but nothing should change :) ## Before submitting - [x] Was this **discussed/approved** via a Github...
## What does this PR do? Wrote unit test for the CIFAR10 dataset and made the dataset import from torchvision to reduce code redundancy as outlined in #839. ## Before...
Hi @otaj - I'm getting the following warnings ``` ========================================================================= warnings summary ========================================================================== ../../../miniconda3/envs/lightning/lib/python3.9/site-packages/torch/utils/tensorboard/__init__.py:4 /home/baruch/miniconda3/envs/lightning/lib/python3.9/site-packages/torch/utils/tensorboard/__init__.py:4: DeprecationWarning: distutils Version classes are deprecated. Use packaging.version instead. if not hasattr(tensorboard, "__version__") or LooseVersion(...