Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Add ckpt_path option to LightningModule.test() #2190

Merged
merged 2 commits into from
Jun 15, 2020
Merged

Add ckpt_path option to LightningModule.test() #2190

merged 2 commits into from
Jun 15, 2020

Commits on Jun 15, 2020

  1. 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.
    @yukw777
    yukw777 committed Jun 15, 2020
    Configuration menu
    Copy the full SHA
    195428a View commit details
    Browse the repository at this point in the history

  2. Update test_set.rst

    @williamFalcon
    williamFalcon committed Jun 15, 2020
    Configuration menu
    Copy the full SHA
    fbfe987 View commit details
    Browse the repository at this point in the history