lightning/tests/models
Jeff Yang 9942f3ebdf
Fix `on_train_batch_start` hook to end epoch early (#3700)
* init

* add test

* changelog and docs

* fix test

* Apply suggestion from code review

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

Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
2020-10-02 21:46:46 +02:00
..
data/horovod part 5 of #3733 (#3774) 2020-10-01 12:34:12 -04:00
__init__.py CI: split tests-examples (#990) 2020-03-25 07:46:27 -04:00
test_amp.py disable optimizers setup during testing (#3059) 2020-09-29 01:09:04 +02:00
test_cpu.py [WIP] ref: deprecated results obj, added support for simpler comms (1/n) (#3681) 2020-09-27 23:19:46 -04:00
test_gpu.py part 5 of #3733 (#3774) 2020-10-01 12:34:12 -04:00
test_grad_norm.py fix gradient norm tracking for row_log_interval > 1 (#3489) 2020-09-15 18:41:27 +02:00
test_hooks.py Fix `on_train_batch_start` hook to end epoch early (#3700) 2020-10-02 21:46:46 +02:00
test_horovod.py Fix global step increment on training_epoch_end (#3673) 2020-09-27 20:19:51 -04:00
test_hparams.py warn user when dropping unpicklable hparams (#2874) 2020-08-28 09:07:43 +02:00
test_onnx.py fix ONNX model save on GPU (#3145) 2020-08-26 16:22:19 +00:00
test_restore.py enable None model checkpoint default (#3669) 2020-09-26 23:14:04 -04:00
test_sync_batchnorm.py updated sync bn (#2838) 2020-08-06 01:12:11 +02:00
test_torchscript.py to_torchscript method for LightningModule (#3258) 2020-09-03 20:24:44 +02:00
test_tpu.py revert backend types (#3788) 2020-10-02 06:18:44 -04:00