Skip to content

Instantly share code, notes, and snippets.

@amankharwal

amankharwal/Linear Regression with PyTorch Secret

Created August 19, 2020 04:20
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 amankharwal/ac9302cda579d9df25247dd3aa50d166 to your computer and use it in GitHub Desktop.
Save amankharwal/ac9302cda579d9df25247dd3aa50d166 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
Linear Regression with PyTorch
inputDim = 1
outputDim = 1
learningRate = 0.01
epochs = 100
model = linearRegression(inputDim, outputDim)
if torch.cuda.is_available():
model.cuda()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment