Skip to content

Instantly share code, notes, and snippets.

@omarsar
Created December 22, 2019 14:14
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save omarsar/125a0d56306263ba4087c9446fca75e4 to your computer and use it in GitHub Desktop.
Save omarsar/125a0d56306263ba4087c9446fca75e4 to your computer and use it in GitHub Desktop.
## test the model
sample = torch.tensor([10.0], dtype=torch.float)
predicted = model(sample)
print(predicted.detach().item())
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment