Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
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
You can’t perform that action at this time.