Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
9,307 Commits 59 Branches 217 Tags 356 MiB
Commit Graph

18 Commits

Author SHA1 Message Date
Adrian Wälchli c712ec1ba9
Add support for saving with full state-dict in Fabric's FSDP (#17526) 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2023-05-11 13:02:30 -04:00
Jirka Borovec 384c203532
ruff: PT some more fixes (#17569) 2023-05-05 08:25:15 +02:00
Jirka Borovec f55d10f5ee
ruff: autofix PT (#17541) 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2023-05-04 11:50:39 -04:00
Adrian Wälchli a533f68693
Support compiling a module after it was set up by Fabric (#17529) 
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2023-05-03 09:00:11 +02:00
pre-commit-ci[bot] 91cb4b9b87
[pre-commit.ci] pre-commit suggestions (#17271) 
* [pre-commit.ci] pre-commit suggestions

updates:
- [github.com/PyCQA/docformatter: v1.4 → v1.6.0](https://github.com/PyCQA/docformatter/compare/v1.4...v1.6.0)
- [github.com/psf/black: 22.12.0 → 23.3.0](https://github.com/psf/black/compare/22.12.0...23.3.0)
- [github.com/charliermarsh/ruff-pre-commit: v0.0.237 → v0.0.260](https://github.com/charliermarsh/ruff-pre-commit/compare/v0.0.237...v0.0.260)

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* update

* apply

* fixing

* docs/lines

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Jirka Borovec <6035284+Borda@users.noreply.github.com>
Co-authored-by: Jirka <jirka.borovec@seznam.cz>
 2023-04-26 21:37:41 +02:00
dependabot[bot] b792c90ea7
Update deepspeed requirement support window (#16813) 
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
Co-authored-by: Jirka <jirka.borovec@seznam.cz>
 2023-04-25 17:26:49 +02:00
Adrian Wälchli d9b4ebd726
Enable precision autocast for `LightningModule` step methods in Fabric (#17439) 2023-04-24 11:50:59 +00:00
Adrian Wälchli 0ee71d6a7a
Fix LightningModule step methods bypassing DDP wrapper in Fabric (#17424) 
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2023-04-21 15:29:32 -04:00
Atharva Phatak ea708da55a
Add `is_wrapped` utility function for Fabric (#16953) 
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
Co-authored-by: awaelchli <aedu.waelchli@gmail.com>
 2023-03-14 13:03:38 +00:00
Adrian Wälchli b6c693d345
Add test for `torch.compile()` with `Fabric.setup()` (#16977) 2023-03-07 10:57:31 -05:00
Jirka Borovec 760612fb8a
update list of fist party packages (#16859) 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2023-03-03 16:55:48 +00:00
Adrian Wälchli 7820a117bc
Optimize precision conversion in forward of Fabric module wrapper (#16903) 2023-03-02 23:41:37 +00:00
Carlos Mocholí 235e692259
Fabric: do `set_epoch` for `batch_sampler.sampler` (#16841) 2023-02-23 00:11:29 +00:00
Jirka Borovec 770b792925
copyright Lightning AI team (#16647) 
* copyright Lightning AI team

* more...
 2023-02-06 15:26:51 +01:00
Jirka Borovec 377210d85d
tests: switch imports for fabric (#16592) 2023-02-01 20:34:38 +00:00
Adrian Wälchli b085fa12d3
Rename leftover definitions in Lite tests (#16309) 2023-01-10 15:02:05 +00:00
Adrian Wälchli c656307127
Handle `set_to_none` when using DeepSpeed optimizer in Lite (#16275) 2023-01-09 09:01:11 -05:00
Carlos Mocholí 15ef52bc73
Rename LightningLite to Fabric (#16244) 
* Rename LightningLite to Fabric

* Fix introspection test

* Fix deprecated Lite tests

* Undo accidental Horovod removal

* Fixes
 2023-01-04 10:57:18 -05:00