37e7582486
* 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> |
||
---|---|---|
.. | ||
data/horovod | ||
__init__.py | ||
test_amp.py | ||
test_cpu.py | ||
test_gpu.py | ||
test_grad_norm.py | ||
test_hooks.py | ||
test_horovod.py | ||
test_hparams.py | ||
test_restore.py |