diff --git a/3_Feedforward_PyTorch/3.3-FFN-Half-Moon.ipynb b/3_Feedforward_PyTorch/3.3-FFN-Half-Moon.ipynb index ae17b867..7c4549ad 100644 --- a/3_Feedforward_PyTorch/3.3-FFN-Half-Moon.ipynb +++ b/3_Feedforward_PyTorch/3.3-FFN-Half-Moon.ipynb @@ -185,7 +185,7 @@ " # in the example below we have 2 input units (num_features) and 2 output units (num_output)\n", " # so our weights become W[2, 2], b[2]\n", " # if we want to make a hidden layer with 100 units, we need to define the shape of the\n", - " # first weight to W[100, 2], b[2] and the shape of the second weight to W[2, 100], b[2]\n", + " # first weight to W[100, 2], b[100] and the shape of the second weight to W[2, 100], b[2]\n", " \n", " # first layer\n", " self.W_1 = nn.Parameter(torch.randn(num_output, num_features)) \n",