lightning/tests/trainer
thomas chaton e76ebd640e
[feat] Add BasePredictionWriter 3/3 (#7127)
* wip

* update

* update

* update

* update

* update

* typo

* update on comments

* update

* update

* update

* update

* update changelog

* update

* Fix merge

* Fix merge

* move code

* resolve test

* add extra test

* add an extra test

* update on comments

* add typing

* resolve flake8

* Refactor and Docs

* Fix tests

* Fix tests

* Fix tests

* Duplicate

* Fix tests

* resolve bug

* update

* update on comments

* Update pytorch_lightning/utilities/imports.py

Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>

* Update pytorch_lightning/utilities/device_parser.py

Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>

* update

* update

* update

* update on comments

* resolve flkae8

* update test

* Apply suggestions from code review

* update on comments

* Update pytorch_lightning/callbacks/prediction_writer.py

Co-authored-by: Ethan Harris <ewah1g13@soton.ac.uk>

* Update pytorch_lightning/callbacks/prediction_writer.py

Co-authored-by: Ethan Harris <ewah1g13@soton.ac.uk>

* Update pytorch_lightning/callbacks/prediction_writer.py

Co-authored-by: Ethan Harris <ewah1g13@soton.ac.uk>

* update on comments

* update

* update on comment

* Apply suggestions from code review

* update

Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
Co-authored-by: Kaushik B <45285388+kaushikb11@users.noreply.github.com>
Co-authored-by: Ethan Harris <ewah1g13@soton.ac.uk>
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
2021-04-27 20:23:55 +00:00
..
connectors Add checkpoint parameter to on_save_checkpoint (#6072) 2021-02-25 21:18:19 +05:30
data_flow Remove evaluation loop legacy dict returns for `*_epoch_end` hooks (#6973) 2021-04-13 12:37:54 +01:00
dynamic_args More EpochResultStore refactors! 🎉 (#5522) 2021-02-11 14:32:45 +00:00
flags Fix validation progress counter with check_val_every_n_epoch > 1 (#5952) 2021-04-02 17:40:41 +09:00
logging_ [feat] Add better support for predict + ddp 2/3 (#7215) 2021-04-27 08:46:45 -04:00
optimization [bugfix] Metric not logged properly in manual optimization (#7228) 2021-04-27 09:16:51 -04:00
properties Refactor PyTorch profiler 4/5 (#6349) 2021-03-23 18:13:29 +01:00
__init__.py split trainer tests (#956) 2020-02-26 20:31:40 -05:00
test_config_validator.py Add trainer.predict config validation (#6543) 2021-03-21 21:07:54 +00:00
test_data_loading.py Auto-set `DataLoader.worker_init_fn` with `seed_everything` (#6960) 2021-04-19 16:28:37 +02:00
test_dataloaders.py [feat] Add better support for predict + ddp 2/3 (#7215) 2021-04-27 08:46:45 -04:00
test_evaluation_loop.py Fix inconsistent outputs in `on_*_end` and `*_end` (#6969) 2021-04-13 15:16:21 +01:00
test_states.py Remove no return warning from val/test step (#6139) 2021-03-06 17:15:21 +00:00
test_supporters.py [bugfix] Add support for CombinedLoader in validation with ddp (#7102) 2021-04-20 08:22:02 +00:00
test_trainer.py [feat] Add BasePredictionWriter 3/3 (#7127) 2021-04-27 20:23:55 +00:00
test_trainer_cli.py fix gpus default for Trainer.add_argparse_args (#6898) 2021-04-09 11:20:43 +02:00
test_trainer_tricks.py missing tests default_root_dir=tmpdir (#6314) 2021-03-04 19:23:12 +00:00
test_training_loop.py Fixed `num_sanity_val_steps` affecting reproducibility of training data shuffling (#7014) 2021-04-27 09:51:39 +00:00