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

325 Commits

Author SHA1 Message Date
jjenniferdai 5d2a3eab69
add `state_dict`/`load_state_dict` to base `Callback` (#11998) 2022-03-04 02:41:48 +00:00
Kaushik B 6ff38d4c8e
Update RichProgressBarTheme after detecting light theme on colab (#10993) 2022-02-22 05:32:27 +00:00
Carlos Mocholí a0ca8d076f
Refactor early stopping test (#11866) 2022-02-17 23:20:39 +00:00
edward-io 87bd54aedf
fix typos (#11937) 2022-02-16 17:27:51 -08:00
Carlos Mocholí 789fae828d
Fix `current_epoch` value on training end (#8578) 
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
 2022-02-10 17:55:59 +01:00
Danielle Pintz 32e7d32956
Delete `test_on_before_accelerator_backend_setup` (#11803) 
Co-authored-by: Carlos Mocholi <carlossmocholi@gmail.com>
 2022-02-08 09:36:37 +00:00
Rohit Gupta 581bf7f2f2
Deprecate `on_epoch_start/on_epoch_end` hook (#11578) 2022-02-07 14:15:27 +00:00
Krishna Kalyan f509e40ae3
Deprecate `on_before_accelerator_backend_setup` callback hook (#11655) 
Co-authored-by: rohitgr7 <rohitgr1998@gmail.com>
 2022-02-07 11:07:21 +00:00
ananthsub a64438c897
Centralize rank_zero_only utilities into their own module (#11747) 
* Centralize rank_zero_only utilities into their own module

Fixes #11746

* PossibleUserWarning

* Update test_warnings.py

* update imports

* more imports

* Update CHANGELOG.md

* Update mlflow.py

* Update cli.py

* Update api_references.rst

* Update meta.py

* add deprecation tests

* debug standalone

* fix standalone tests

* Update CHANGELOG.md
 2022-02-07 08:09:55 +00:00
Dan Dale 3bc2407239
Allow access to ckpt_path within context of fit() (#11696) 
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
 2022-02-05 05:23:16 +01:00
Rohit Gupta 01abe72278
Fix to avoid val progress bar disappear after validate (#11700) 
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
 2022-02-03 13:35:38 +00:00
Rohit Gupta e9065e9d42
Fix rich with uneven refresh rate tracking (#11668) 
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
 2022-02-03 10:27:05 +00:00
Danielle Pintz 9ebd7df22a
Move progress bar disabling out of the Trainer (#11377) 
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
 2022-02-03 06:29:32 +00:00
Rohit Gupta eceefdc602
Fix rich progress bar render only on main pbar (#11690) 2022-02-03 04:18:07 +00:00
Krishna Kalyan 6291af5c19
Replace occurrences of `on_before_accelerator_backend_setup_called` with `setup` (#11568) 
Co-authored-by: Carlos Mocholi <carlossmocholi@gmail.com>
 2022-02-03 04:14:33 +00:00
Rohit Gupta ee049e123d
Fix rich progress bar metric render on epoch end (#11689) 
Co-authored-by: Carlos Mocholi <carlossmocholi@gmail.com>
Co-authored-by: Jirka <jirka.borovec@seznam.cz>
 2022-02-03 01:43:48 +00:00
Carlos Mocholí 62818dbace
Use a dataclass as the scheduler config (#11443) 2022-01-18 20:23:32 +01:00
Jv Kyle Eclarin c85946531d
Update `tests/callbacks/*.py` to use `devices` instead of `gpus` or `ipus` (#11387) 
Co-authored-by: Kaushik B <45285388+kaushikb11@users.noreply.github.com>
 2022-01-18 19:13:01 +05:30
Aki Nitta 8dc36c3745
Fix inconsistent exceptions raised with no `rich` installed (#11360) 
Co-authored-by: Carlos Mocholi <carlossmocholi@gmail.com>
 2022-01-12 03:55:51 +00:00
Carlos Mocholí 9771040621
Add typing to `TQDMProgressBar` (#11369) 2022-01-12 01:07:30 +00:00
Jv Kyle Eclarin 3e0569fccc
Update test_pruning.py to use `devices` instead of `gpus` or `ipus` (#11341) 2022-01-09 17:15:29 +09:00
Jv Kyle Eclarin b56d8677ad
Update test_pruning.py to use `devices` instead of `gpus` or `ipus` (#11339) 2022-01-08 17:24:29 +01:00
Jv Kyle Eclarin 4710a8128b
Update test_gpu_stats_monitor.py to use `devices` instead of `gpus` or `ipus` (#11340) 2022-01-08 01:38:25 +00:00
Carlos Mocholí 8a549a550c
Integrate progress tracking into the progress bar (#11213) 2022-01-06 14:29:48 +01:00
ananthsub 05ed9a201c
Group metrics generated by `DeviceStatsMonitor` for better visualization (#11254) 
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
 2022-01-03 13:26:17 +00:00
Adrian Wälchli c210e338ef
Update strategy import statements (#11231) 2021-12-23 08:26:28 +01:00
Kaushik B 576a5d62a0
Introduce strategies directory for Training Strategies (#11226) 
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
 2021-12-22 20:23:30 +00:00
four4fish cf5ef32f7b
Deprecate Trainer.training_type_plugin in favor of trainer.strategy (#11141) 
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
 2021-12-22 02:11:43 +00:00
four4fish f98cd78e9e
Renamed the `DDPSpawnPlugin` to `DDPSpawnStrategy` (#11145) 2021-12-21 23:06:14 +00:00
Adrian Wälchli 29eb9cccf2
Rename the `TrainingTypePlugin` base to `Strategy` (#11120) 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>
Co-authored-by: four4fish <88516121+four4fish@users.noreply.github.com>
 2021-12-20 12:50:11 +00:00
Rohit Gupta 3461af0ddb
Add support for returning callback from `LightningModule.configure_callbacks` (#11060) 2021-12-18 10:46:35 +00:00
Rohit Gupta 860959fb3f
Enable logging hparams only if there are any (#11105) 2021-12-17 19:40:56 +01:00
Adrian Wälchli 978f5e6ad6
Fix AttributeError when using CombinedLoader in prediction (#11111) 
Co-authored-by: Carlos Mocholi <carlossmocholi@gmail.com>
 2021-12-17 18:02:25 +00:00
Carlos Mocholí 7e10f6d41f
Save the loop progress state by default (#10784) 2021-12-17 16:00:27 +00:00
Carlos Mocholí 3c4d06bd42
Update the TQDM progress bar `on_train_epoch_end` (#11069) 2021-12-15 17:48:32 +00:00
Kaushik B 5eecdcae87
Fix sanity check for RichProgressBar (#10913) 2021-12-15 09:23:35 +00:00
Danielle Pintz 01f5f99919
Deprecate callback hooks `on_init_start` and `on_init_end` (#10940) 2021-12-08 07:42:19 +00:00
Adrian Wälchli c55bc433ce
Fix retrieval of batch indices when dataloader num_workers > 0 (#10870) 
Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>
 2021-12-02 10:36:10 +00:00
Aka.Fido 72cc8b7ca9
Disable validation completely when `overfit_batches>0` (#9709) 
Co-authored-by: thomas chaton <thomas@grid.ai>
Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2021-12-01 13:57:57 +00:00
Kaushik B ec0fb2fd95
Raise exception if rich is less than 10.2.2 (#10839) 2021-12-01 06:14:19 +00:00
four4fish 1d2878523a
2/n Move Precision Plugin into strategy - move optimizer related logics (#10596) 
Co-authored-by: Danielle Pintz <38207072+daniellepintz@users.noreply.github.com>
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
Co-authored-by: thomas chaton <thomas@grid.ai>
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
 2021-11-30 08:31:23 +00:00
Carlos Mocholí 724a92b065
Mark outputs as protected in the evaluation loops (#10781) 
Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>
 2021-11-28 20:09:30 +00:00
Carlos Mocholí 152eb57def
Rename special to standalone (#10779) 2021-11-26 17:13:14 +00:00
Kaushik B 48cf1adfd3
Move Colab setup to ProgressBar (#10542) 
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 2021-11-23 06:16:31 +00:00
Kaushik B 137b62d80d
Add `refresh_rate` to RichProgressBar (#10497) 
Co-authored-by: ananthsub <ananth.subramaniam@gmail.com>
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
 2021-11-19 05:59:57 +00:00
Carlos Mocholí 0fa07da987
Fail the test when a `DeprecationWarning` is raised (#9940) 2021-11-17 23:41:50 +01:00
Carlos Mocholí ba036fdeea
Support special test parametrizations (#10569) 2021-11-17 15:46:14 +00:00
Carlos Mocholí 6dfcb6afc5
Skip strategy=ddp_spawn, accelerator=cpu, python>=3.9 tests (#10550) 2021-11-16 10:06:47 +05:30
Carlos Mocholí dcafc95f2b
Avoid deprecated `progress_bar_refresh_rate` usage (#10520) 
Co-authored-by: Danielle Pintz <38207072+daniellepintz@users.noreply.github.com>
 2021-11-15 22:04:48 +01:00
Carlos Mocholí 7a9a08c5d3
Drop torch 1.6 testing (#10390) 
* Drop torch 1.6 support

* Drop 1.6 support

* Update CHANGELOG

* Fixes

* Split change

* Undo change

* 1.7 -> 1.7.1

https://github.com/pytorch/pytorch/issues/47354

* Force trigger nightly

* Update .github/workflows/events-nightly.yml

Co-authored-by: Aki Nitta <nitta@akihironitta.com>

* Revert 1.7.1 change - try wildcard

* Update adjust versions and test it

* Undo test changes

* Revert "Undo test changes"

This reverts commit 3a6acadd11.

* Update CHANGELOG.md

Co-authored-by: Aki Nitta <nitta@akihironitta.com>
 2021-11-13 20:35:03 +00:00