Skip to content

Loading a checkpoint to make predictions #1842

Answered by isaaccorley
lcoandrade asked this question in Q&A

You must be logged in to vote

When you load the checkpoint, you need to pass map_location="cpu" like

task = SemanticSegmentationTask.load_from_checkpoint(ckpt_path, map_location="cpu")

See for more info https://pytorch.org/docs/stable/generated/torch.load.html

Replies: 3 comments 2 replies

You must be logged in to vote
0 replies

You must be logged in to vote
1 reply
@lcoandrade

Answer selected by lcoandrade

You must be logged in to vote
1 reply
@lcoandrade

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
trainers PyTorch Lightning trainers
3 participants