Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
# get predictions for test data
with torch.no_grad():
preds = model(test_seq.to(device), test_mask.to(device))
preds = preds.detach().cpu().numpy()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment