lightning/tests/base
Peter Yu 37e7582486
Add ckpt_path option to LightningModule.test() (#2190)
* Add ckpt_path option to LightningModule.test()

If ckpt_path is "best" (default), it loads the best weights saved by ModelCheckpoint for the test loop.
If ckpt_path is a path to a checkpoint file, it loads the weights from the file for the test loop.
If ckpt_path is None, it uses the weights from the end of training for the test loop.
If model parameter is set, ckpt_path is ignored.

* Update test_set.rst

Co-authored-by: William Falcon <waf2107@columbia.edu>
2020-06-15 08:02:37 -04:00
..
__init__.py Tests: refactor cleanup (#1744) 2020-05-10 13:15:28 -04:00
dataloaders.py Tests: refactor cleanup (#1744) 2020-05-10 13:15:28 -04:00
datasets.py default test logger (#1478) 2020-04-21 20:33:10 -04:00
mixins.py Fix lr key name in case of param groups (#1719) 2020-05-10 17:05:34 -04:00
model_optimizers.py replace Hparams by init args (#1896) 2020-05-24 18:59:08 -04:00
model_template.py update hparams, allow OmegaConf (#2047) 2020-06-08 07:19:34 -04:00
model_test_dataloaders.py Tests: refactor cleanup (#1744) 2020-05-10 13:15:28 -04:00
model_test_epoch_ends.py Tests: refactor cleanup (#1744) 2020-05-10 13:15:28 -04:00
model_test_steps.py Tests: refactor cleanup (#1744) 2020-05-10 13:15:28 -04:00
model_train_dataloaders.py Tests: refactor cleanup (#1744) 2020-05-10 13:15:28 -04:00
model_train_steps.py hotfix to unblock hparams and OmniConf - removes auto_register_init_args by default (#2025) 2020-05-31 08:29:51 -04:00
model_utilities.py Replaces ddp .spawn with subprocess (#2029) 2020-06-01 11:00:32 -04:00
model_valid_dataloaders.py Tests: refactor cleanup (#1744) 2020-05-10 13:15:28 -04:00
model_valid_epoch_ends.py Add ckpt_path option to LightningModule.test() (#2190) 2020-06-15 08:02:37 -04:00
model_valid_steps.py hotfix to unblock hparams and OmniConf - removes auto_register_init_args by default (#2025) 2020-05-31 08:29:51 -04:00
models.py replace Hparams by init args (#1896) 2020-05-24 18:59:08 -04:00
utils.py cleaning (#2030) 2020-06-04 11:25:07 -04:00