Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
#load weights of best model
path = 'saved_weights.pt'
model.load_state_dict(torch.load(path))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment