Update 301_regression.ipynb
This commit is contained in:
@ -125,7 +125,7 @@
|
||||
},
|
||||
"outputs": [],
|
||||
"source": [
|
||||
"optimizer = torch.optim.SGD(net.parameters(), lr=0.4)\n",
|
||||
"optimizer = torch.optim.SGD(net.parameters(), lr=0.2)\n",
|
||||
"loss_func = torch.nn.MSELoss() # this is for regression mean squared loss"
|
||||
]
|
||||
},
|
||||
|
||||
Reference in New Issue
Block a user