Commit Graph

58 Commits

Author SHA1 Message Date
Adrian Wälchli 18106a8f95
Split train- and val progress into separate bars (#16695) 2023-02-09 19:43:50 +00:00
Jirka Borovec 560d6d7956
tests: drop slow flag (#16692)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-02-08 17:09:59 +00:00
Carlos Mocholí bf51844917
Remove memory-retaining epoch-end hooks (#16520) 2023-02-06 17:00:36 +00:00
Adrian Wälchli cd0eedb082
Set `find_unused_parameters=False` as the default (#16611) 2023-02-06 16:51:21 +01:00
Jirka Borovec 770b792925
copyright Lightning AI team (#16647)
* copyright Lightning AI team

* more...
2023-02-06 15:26:51 +01:00
Jirka Borovec de005321a5
tests: switch imports for pytorch (#16595)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-02-02 10:06:45 +00:00
Jirka Borovec 377210d85d
tests: switch imports for fabric (#16592) 2023-02-01 20:34:38 +00:00
Carlos Mocholí ef2a6088ff
Drop support for PyTorch 1.10 (#16492)
* Drop support for PyTorch 1.10

* CHANGELOG

* READMEs

* mypy

* ls

* New poplar version

* Fixed tests

* links

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

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

* skip azure badges

* Table

* Matching dockerfiles

* Drop unnecessary channels and packages

* Push nightly

* Undo unrelated changes

* Revert "Push nightly"

This reverts commit 9618f737c4.

---------

Co-authored-by: Jirka <jirka.borovec@seznam.cz>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-02-01 14:09:12 -05:00
Adrian Wälchli 6a56586492
Make manual optimization mandatory for multiple optimizers (#16539)
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
2023-02-01 16:21:01 +00:00
Carlos Mocholí df09370827
Run `on_train_epoch_end` after the LM for callbacks that monitor (#16567) 2023-02-01 15:27:16 +01:00
Adrian Wälchli 8fc4fb18e6
Switch multi-optimizer tests to manual optimization (#16559) 2023-01-30 17:18:37 +00:00
Carlos Mocholí d78cf99176
Remove the "native" suffix from the codebase (#16490) 2023-01-25 14:09:09 +00:00
Carlos Mocholí 256199ff7c Remove support for logging multiple metrics together (#16389) 2023-01-19 18:39:36 +01:00
Adrian Wälchli cf0952b25e Remove special handling of loss in progress bar (#16192)
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
2023-01-19 18:39:36 +01:00
Adrian Wälchli 03a699693b Remove truncated backpropagation from loops (#16337)
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
2023-01-19 18:39:36 +01:00
Carlos Mocholí 0f4f809a7c
Deprecate the FairScale integration (#16353) 2023-01-17 02:29:06 +01:00
Mayanand d206ca3dab
added support for logging in different trainer stages (#16002)
Co-authored-by: thinkin-machine <you@example.com>
Co-authored-by: Jirka Borovec <6035284+Borda@users.noreply.github.com>
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
2023-01-09 10:33:13 -05:00
Carlos Mocholí 0a928e8ead
Reuse code in demos.BoringModel (#16242) 2023-01-06 09:47:08 +01: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
Carlos Mocholí a58639ce7e
Nightly PyTorch version is now 2.0 (#16017) 2022-12-14 12:37:00 +01:00
Carlos Mocholí 53ceb156c4
Integrate lightning_utilities==0.4.2 (#15817)
Co-authored-by: Jirka Borovec <6035284+Borda@users.noreply.github.com>
2022-12-13 13:13:51 +00:00
Carlos Mocholí 9ed43c64b6
Set the logger explicitly in tests (#15815) 2022-12-10 13:32:37 +00:00
Dan Dale 9a4e8a8c52
Enhance `reduce_boolean_decision` to accommodate `any`-analogous semantics expected by `EarlyStopping` Callback (#15253)
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
2022-11-09 20:17:42 +00:00
Sitcebelly 94bed87a34
Implement freeze batchnorm with freezing track running stats (#15063)
Co-authored-by: awaelchli <aedu.waelchli@gmail.com>
2022-11-01 16:11:42 +00:00
Rohit Gupta 61ae35c378
Use sklearn in runif (#15426)
* Use sklearn in runif
* test by removing sklearn dep
* remove repeated code
* seed
2022-11-01 11:40:32 +00:00
Rohit Gupta 773cb3e8c8
Fix skipped tests due to sklearn (#15311)
Co-authored-by: awaelchli <aedu.waelchli@gmail.com>
2022-10-31 13:58:34 +05:30
Adrian Wälchli 6b0d41cb8a
Fix issues when RichProgressBar disabled (#15376) 2022-10-29 00:52:35 +00:00
Adrian Wälchli 5eafa52596
Fix resetting internal bars in RichProgressBar after each trainer stage (#15377) 2022-10-28 06:20:45 -04:00
Adrian Wälchli ac89d70d4a
Fix pickling issues with rich progress bar (#15319) 2022-10-26 15:25:11 +00:00
Atharva Phatak 4322f53874
Update mypy version (#15161)
* update mypy version
* type-ignore-comments
* more mypy-fix
* import-fix
* Update Lite too
* simpler implementation for flatten dict
* Fix rich progress
* Simplify rich test
* True None

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
2022-10-22 23:26:43 +02:00
Carlos Mocholí c334b7766c
Remove old testing artifacts (#15052) 2022-10-10 17:34:18 +00:00
Adrian Wälchli 3183079204
Remove deprecated callback hooks (#14834)
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
Co-authored-by: otaj <ota@lightning.ai>
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
2022-10-10 15:46:28 +00:00
Adrian Wälchli 8f90084059
Remove deprecated on_load/save_checkpoint behavior (#14835)
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
2022-10-10 11:08:13 +00:00
otaj 7e518cacd2
Use `torch.testing.assert_close` everywhere (#15031)
remove unnecessary version check
2022-10-07 16:59:04 +02:00
Carlos Mocholí 7ef87464dd
Refactor XLA and TPU checks across codebase (#14550) 2022-10-04 22:54:14 +00:00
Andres Algaba 3daa4c9cc0
Remove deprecated on_init_start_end (#14867)
Co-authored-by: awaelchli <aedu.waelchli@gmail.com>
Co-authored-by: Akihiro Nitta <nitta@akihironitta.com>
Co-authored-by: otaj <6065855+otaj@users.noreply.github.com>
2022-09-30 15:11:38 +00:00
Adrian Wälchli 31788db337
Fix attribute error in SWA when running with Tuner (#14836)
* Fix attribute error in SWA when running with Tuner
* changelog
* add better test

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-09-22 07:09:59 +00:00
Rohit Gupta 8c6119fbce
Add auto wrapping support for `DDPFullyShardedStrategy` (#14383) 2022-09-05 19:07:26 +00:00
Carlos Mocholí e0c2c3e677
Clean up fairscale imports (#14476) 2022-09-01 18:08:40 +02:00
Adrian Wälchli 28e18881a9
Mark stage argument in hooks as required (#14064)
Co-authored-by: rohitgr7 <rohitgr1998@gmail.com>
2022-09-01 15:47:40 +02:00
Carlos Mocholí 291267c3bf
Unify rank zero messaging utilities (#14116) 2022-08-30 09:51:30 +00:00
Rohit Gupta f3574176e2
Change `trainer.should_stop` to not stop in between an epoch and run until `min_steps/min_epochs` only (#13890) 2022-08-27 12:12:24 +00:00
Kaushik B a8c6e69b43
Fix wrong num padding for RichProgressBar (#14296) 2022-08-19 09:40:44 +05:30
Carlos Mocholí 9b61b1c482
Remove duplicated test classes (#14122)
Remove duplicated classes
2022-08-10 17:21:05 +02:00
Adam Reeve 975a4fc2f1
Support checkpoint save and load with Stochastic Weight Averaging (#9938)
Co-authored-by: thomas chaton <thomas@grid.ai>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
Co-authored-by: Carlos Mocholi <carlossmocholi@gmail.com>
Co-authored-by: Kushashwa Ravi Shrimali <kushashwaravishrimali@gmail.com>
Co-authored-by: Jirka <jirka.borovec@seznam.cz>
Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>
2022-08-09 23:18:21 +00:00
Adrian Wälchli fff62f0ae5
Fix TPU testing and collect all tests (#11098)
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: Jirka Borovec <Borda@users.noreply.github.com>
Co-authored-by: Kaushik B <45285388+kaushikb11@users.noreply.github.com>
2022-07-27 15:40:40 +00:00
Carlos Mocholí 9f51c07604
Support setting the trainer reference recursively for ensembles (#13638)
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
Co-authored-by: Justus Schock <12886177+justusschock@users.noreply.github.com>
2022-07-22 19:58:46 +02:00
Rohit Gupta e451fa28d0
Fix main progress bar counter when `val_check_interval=int` and `check_val_every_n_epoch=None` (#12832) 2022-07-20 20:33:00 +00:00
Justus Schock c75457da99 Rename GPUAccelerator to CUDAAccelerator 2022-07-19 13:06:30 -04:00
George Stein 0449e861cc
Fix `trainer.predict(return_predictions=False)` does not track `batch_indices` (#13629)
* Pull request for fixing issue #13580
* chlog and test
* disable track for epoch

Co-authored-by: rohitgr7 <rohitgr1998@gmail.com>
2022-07-18 08:26:15 +00:00