lightning-bolts
lightning-bolts copied to clipboard
Toolbox of models, callbacks, and datasets for AI/ML researchers.
The pre-trained weights for a cifar-10 SimCLR model seem wrong on line: 327 of docs. The weights say that they are imagnet pretrained but are provided under cifar-10.
## 📚 Documentation The documentation for [pl_bolts/metrics/aggregation.py](https://github.com/Lightning-AI/lightning-bolts/blob/master/pl_bolts/metrics/aggregation.py) seems to be missing and `accuracy` is missing its docstring. I assume that it should look more like https://github.com/Lightning-AI/lightning-bolts/blob/master/pl_bolts/metrics/object_detection.py. I don't think that...
## What does this PR do? Part of https://github.com/Lightning-AI/lightning-bolts/issues/839 `pl_bolts.models.gans.pix2pix.components.DownSampleConv` `pl_bolts.models.gans.pix2pix.components.Generator` `pl_bolts.models.gans.pix2pix.components.PatchGAN` `pl_bolts.models.gans.pix2pix.components.UpSampleConv` `pl_bolts.models.gans.pix2pix.pix2pix_module._weights_init` `pl_bolts.models.gans.pix2pix.pix2pix_module.Pix2Pix` ## Before submitting - [x] Was this **discussed/approved** via a Github issue? (no need for...
## 🚀 Feature Support lightning 2.* and pytorch 2.* Is there a time plan? ### Motivation Pytorch lightning 2. and pytorch 2.* are both available and with nice new and...
## 🚀 Feature I think it would be great to integrate the current [RL bolts models](https://github.com/Lightning-AI/lightning-bolts/tree/master/pl_bolts/models/rl) with the new [TorchRL](https://github.com/pytorch/rl) library ### Motivation While `gym` is a great library for...
updates: - [github.com/pre-commit/pre-commit-hooks: v4.4.0 → v4.6.0](https://github.com/pre-commit/pre-commit-hooks/compare/v4.4.0...v4.6.0) - [github.com/asottile/pyupgrade: v3.8.0 → v3.16.0](https://github.com/asottile/pyupgrade/compare/v3.8.0...v3.16.0) - [github.com/PyCQA/docformatter: v1.7.3 → v1.7.5](https://github.com/PyCQA/docformatter/compare/v1.7.3...v1.7.5) - [github.com/executablebooks/mdformat: 0.7.16 → 0.7.17](https://github.com/executablebooks/mdformat/compare/0.7.16...0.7.17) - [github.com/psf/black: 23.3.0 → 24.4.2](https://github.com/psf/black/compare/23.3.0...24.4.2) - [github.com/astral-sh/ruff-pre-commit: v0.0.276 →...
Bumps [Lightning-AI/utilities](https://github.com/lightning-ai/utilities) from 0.9.0 to 0.10.1. Release notes Sourced from Lightning-AI/utilities's releases. Minor patch 0.10.1 What's Changed Fixed Avoid accidental namedtuple conversion in apply_to_collection (#210) Full Changelog: https://github.com/Lightning-AI/utilities/compare/v0.10.0...v0.10.1 New Contributors...
Bumps [actions/upload-artifact](https://github.com/actions/upload-artifact) from 3 to 4. Release notes Sourced from actions/upload-artifact's releases. v4.0.0 What's Changed The release of upload-artifact@v4 and download-artifact@v4 are major changes to the backend architecture of Artifacts....
Updates the requirements on [torchmetrics](https://github.com/Lightning-AI/torchmetrics) to permit the latest version. Release notes Sourced from torchmetrics's releases. New Image metrics & wrappers [1.3.0] - 2024-01-10 Added Added more tokenizers for SacreBLEU...
I use images with resolutions of 128x128 and 256x256 as inputs for the VAE, but the reconstructed images are of very poor quality, typically only reconstructing 4x4 image blocks. Is...