Update Part 2 - Neural Networks in PyTorch (Solution).ipynb

Sisim · web-flow · commit 5369c2941289 · 2019-09-25T16:53:14.000+02:00
activation function for the output is maybe forgotten.
diff --git a/intro-to-pytorch/Part 2 - Neural Networks in PyTorch (Solution).ipynb b/intro-to-pytorch/Part 2 - Neural Networks in PyTorch (Solution).ipynb
@@ -172,7 +172,7 @@
     "\n",
     "h = activation(torch.mm(inputs, w1) + b1)\n",
     "\n",
-    "out = torch.mm(h, w2) + b2"
+    "out = activation(torch.mm(h, w2) + b2)"
    ]
   },
   {

Original file line number	Diff line number	Diff line change
`@@ -172,7 +172,7 @@`
`172`	`172`	`"\n",`
`173`	`173`	`"h = activation(torch.mm(inputs, w1) + b1)\n",`
`174`	`174`	`"\n",`
`175`		`- "out = torch.mm(h, w2) + b2"`
	`175`	`+ "out = activation(torch.mm(h, w2) + b2)"`
`176`	`176`	`]`
`177`	`177`	`},`
`178`	`178`	`{`