Skip to content

what's the difference between load_from_checkpoint and resume_from_checkpoint #11378

Discussion options

You must be logged in to vote

resume_from_checkpoint is used to resume the training using the checkpointed state_dicts. It will reload model's state_dict, optmizer's and schedulers's state_dicts, training state as well in a general case.
use-case: to restart the training

load_from_checkpoint just reloads the model's state_dict and return the model with the loaded weights.
use-case: for quick evaluation/prediction.

Replies: 1 comment 1 reply

Comment options

You must be logged in to vote
1 reply
@marsggbo
Comment options

Answer selected by marsggbo
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment