lightning/tests/models
Kaushik B 5dfd62c09e Disable training with zero num_training_batches when insufficient limit_train_batches (#5703)
* disable training when zero num_train_batches with limit_train_batches

* refactor train skip condition

* fix formatting issues

* fix formatting issues

* ref: test error msg

* fix tests for data loader calls

* fix train dataloader condition

* update limit_train_batches upper range in test comment

* remove model state check test

Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com>
2021-02-05 21:40:42 +01:00
..
conf bugfix: Resolve interpolation bug with Hydra (#5406) 2021-01-26 14:29:47 +01:00
data Disable training with zero num_training_batches when insufficient limit_train_batches (#5703) 2021-02-05 21:40:42 +01:00
__init__.py
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 fix tests 2021-02-04 20:55:58 +01:00
test_horovod.py Fix ModelCheckpoint race condition in file existence check (#5155) 2021-02-05 21:40:39 +01: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