Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
lightning/tests/helpers
History
shuyingsunshine21 299f2c481b
FSDP with full state dict (#7487) 
* Fix some test errors
Summary:

Test Plan:

Reviewers:

Subscribers:

Tasks:

Tags:

* checkpoint consolidation

* Update ddp_spawn.py

* Update test_metric_result_integration.py

* Update test_results.py

* Update utils.py

* Update utils.py

* Update test_all_gather_grad.py

* Update test_all_gather_grad.py

* Update test_results.py

* Revert "Update test_results.py"

This reverts commit 9d4a2b891d.

* Revert "Merge pull request #1 from shuyingsunshine21/shuyingsunshine21-checkpoint_consolidate"

This reverts commit c5053da789, reversing
changes made to 0d23d75bc9.

* Revert "Update test_all_gather_grad.py"

This reverts commit 0d23d75bc9.

* Revert "Update utils.py"

This reverts commit 70fe5da9c6.

* Revert "Update utils.py"

This reverts commit a9aae99f6e.

* Revert "Update test_results.py"

This reverts commit ea74906878.

* Revert "Update test_metric_result_integration.py"

This reverts commit bf70e431b3.

* Revert "Update ddp_spawn.py"

This reverts commit f17210183b.

* Revert "checkpoint consolidation"

This reverts commit 536c1323b0.

* Revert "Revert "checkpoint consolidation""

This reverts commit 3a9fde915a.

* Revert "Revert "Revert "checkpoint consolidation"""

This reverts commit 7a369f47e1.

* Revert "Revert "Update ddp_spawn.py""

This reverts commit 8222dc98ea.

* Revert "Revert "Update test_metric_result_integration.py""

This reverts commit 6c095b2370.

* Revert "Revert "Update test_results.py""

This reverts commit 250d0aaaa2.

* Revert "Revert "Update utils.py""

This reverts commit 8651d54d79.

* Revert "Revert "Update test_all_gather_grad.py""

This reverts commit dcdcd29731.

* modify distributed environment to make test pass

* fix version for ddp plugin test

* fix

* fix

* changelog

* Update CHANGELOG.md

* fsdp with full state dict

* fix missing import

* modify unitest

* fix

* fix

* fix typo

* modify test and add changelog

* fix

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

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

* limit max_epoch to 1 for testing

* test

* fix

* update

* testing remove special for multi gpu

* assert gpu

* add assertion for gpu

* fix

* Re-enable special test, use ModelCheckpoint

* Fix paths

* Fix path passing

* test

* test

* fix test

* fix

* pre-commit format

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

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

Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: SeanNaren <sean@grid.ai>
 		2021-05-24 08:11:45 +01:00
..
__init__.py fix miss-leading imports in tests (#5873) 2021-02-09 05:10:52 -05:00
advanced_models.py fixing examples (#6600) 2021-03-20 18:58:59 +00:00
boring_model.py [bugfix] Fix dataloading for iterable datasets and limit_train_batches (#7306) 2021-05-03 19:50:26 +01:00
dataloaders.py
…
datamodules.py Delete tests.helpers.TrialMNISTDataModule (#5999) 2021-02-18 03:35:38 +00:00
datasets.py fixing examples (#6600) 2021-03-20 18:58:59 +00:00
deterministic_model.py Remove legacy support for the magic `log`/`progress_bar` keys in dict returns (#6734) 2021-03-31 00:28:04 +02:00
imports.py fix importing torchtext batch (#6365) 2021-03-05 21:39:52 +01:00
pipelines.py `TrainerState` refactor [5/5] (#7173) 2021-05-04 12:50:56 +02:00
runif.py FSDP with full state dict (#7487) 2021-05-24 08:11:45 +01:00
simple_models.py update lr_finder to check for attribute if not running fast_dev_run (#5990) 2021-02-17 07:15:29 -05:00
test_datasets.py Add PyTorch 1.8 Profiler 5/5 (#6618) 2021-03-23 20:43:21 +00:00
test_models.py fix miss-leading imports in tests (#5873) 2021-02-09 05:10:52 -05:00
utils.py Remove no return warning from val/test step (#6139) 2021-03-06 17:15:21 +00:00