lightning/tests/trainer
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 split trainer tests (#956) 2020-02-26 20:31:40 -05:00
test_checks.py replace Hparams by init args (#1896) 2020-05-24 18:59:08 -04:00
test_dataloaders.py Add ckpt_path option to LightningModule.test() (#2190) 2020-06-15 08:02:37 -04:00
test_lr_finder.py remove deprecated API for v0.8 (#2073) 2020-06-12 14:37:52 -04:00
test_optimizers.py remove deprecated API for v0.8 (#2073) 2020-06-12 14:37:52 -04:00
test_trainer.py Add ckpt_path option to LightningModule.test() (#2190) 2020-06-15 08:02:37 -04:00
test_trainer_cli.py handle unknown args passed to Trainer.from_argparse_args (#1932) 2020-05-25 16:01:29 -04:00
test_trainer_tricks.py remove deprecated API for v0.8 (#2073) 2020-06-12 14:37:52 -04:00