lightning/tests/models
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
..
data/horovod RC & Docs/changelog (#1776) 2020-05-11 21:57:53 -04:00
__init__.py
test_amp.py Tests: refactor cleanup (#1744) 2020-05-10 13:15:28 -04:00
test_cpu.py update hparams, allow OmegaConf (#2047) 2020-06-08 07:19:34 -04:00
test_gpu.py Replaces ddp .spawn with subprocess (#2029) 2020-06-01 11:00:32 -04:00
test_grad_norm.py Mistake in parameters' grad norm tracking (#2012) 2020-06-02 18:51:09 -04:00
test_hooks.py data transfer model hook (+ refactor) (#1756) 2020-06-02 21:45:19 -04:00
test_horovod.py replace Hparams by init args (#1896) 2020-05-24 18:59:08 -04:00
test_hparams.py past checkpoints (#2160) 2020-06-14 11:36:45 -04:00
test_restore.py Add ckpt_path option to LightningModule.test() (#2190) 2020-06-15 08:02:37 -04:00