Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
lightning/tests/models
History
Swetha Mandava c62f68c7cd passing batch outputs to on_train_batch_end (#4369) 
* passing batch outputs to on_train_batch_end

* styling

* updating epoch end logic

* also condition on on_train_epoch_end hooks

* more readable

* pep8

* pep8

* readability suggestion accepted

Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>

* adding test_training_epoch_end_metrics_collection_on_override test

* fix formatting

* fix formatting

Co-authored-by: Swetha Mandava <smandava@nvidia.com>
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
Co-authored-by: Sean Naren <sean.narenthiran@gmail.com>
Co-authored-by: Roger Shieh <sh.rog@protonmail.ch>

(cherry picked from commit 5fcca4e43b)
 		2021-02-04 20:55:41 +01:00
..
conf bugfix: Resolve interpolation bug with Hydra (#5406) 2021-01-26 14:29:47 +01:00
data/horovod Fix pre-commit isort failure on tests/models/*.py (#5423) 2021-01-14 09:42:01 -05:00
__init__.py CI: split tests-examples (#990) 2020-03-25 07:46:27 -04:00
test_amp.py fix reinit_schedulers with correct optimizer (#5519) 2021-02-03 19:41:45 +01:00
test_cpu.py fix imports / isort / flake8 2021-01-26 14:57:34 +01:00
test_gpu.py Refactor: legacy accelerators and plugins (#5645) 2021-01-26 20:04:36 -05:00
test_grad_norm.py prune check on Trainer fit result (#5453) 2021-01-11 19:36:48 -05:00
test_hooks.py passing batch outputs to on_train_batch_end (#4369) 2021-02-04 20:55:41 +01:00
test_horovod.py Refactor: legacy accelerators and plugins (#5645) 2021-01-26 20:04:36 -05:00
test_hparams.py fix imports / isort / flake8 2021-01-26 14:57:34 +01:00
test_model_hooks.py simple tests restructure (#5452) 2021-01-15 20:58:20 -05:00
test_onnx.py Allow any input in to_onnx and to_torchscript (#4378) 2020-12-12 18:17:03 +08:00
test_restore.py Force ModelCheckpoint callback to run last (#5731) 2021-02-03 16:40:57 -05:00
test_sync_batchnorm.py Refactor: legacy accelerators and plugins (#5645) 2021-01-26 20:04:36 -05:00
test_torchscript.py Refactor setup_training and remove test_mode (#5388) 2021-02-03 19:41:42 +01:00
test_tpu.py Fix pre-commit isort failure on tests/models/*.py (#5423) 2021-01-14 09:42:01 -05:00