Skip to content

Commit dbb93c3

Browse files
committed
Small changes
1 parent f2f098b commit dbb93c3

File tree

1 file changed

+1
-8
lines changed

1 file changed

+1
-8
lines changed

exercises/01_penguin_classification.ipynb

Lines changed: 1 addition & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -342,7 +342,7 @@
342342
"\n",
343343
"  The module `torch.nn` contains different classes that help you build neural network models. All models in PyTorch inherit from the subclass `nn.Module`, which has useful methods like `parameters()`, `__call__()` and others.\n",
344344
"\n",
345-
"  `torch.nn` also has various layers that you can use to build your neural network. For example, we will use `nn.Linear` in our code below, which constructs a fully connected layer. In particular, we will two `nn.Linear` layers as part of our network in the `__init__()` method. `torch.nn.Linear` is a subclass of `torch.nn.Module`. \n",
345+
"  `torch.nn` also has various layers that you can use to build your neural network. For example, we will use `nn.Linear` in our code below, which constructs a fully connected layer. `torch.nn.Linear` is a subclass of `torch.nn.Module`. \n",
346346
"\n",
347347
"  What exactly is a \"layer\"? It is essentially a step in the neural network computation. i.e. The `nn.Linear` layer computes the linear transformation of the input vector `$x$`: `$y$ = $W^T x + b$`. Where `W` is the matrix of tunable parameters and `b` is a bias vector.\n",
348348
"\n",
@@ -384,13 +384,6 @@
384384
"# define a model and print and test (try with torch.rand() function)"
385385
]
386386
},
387-
{
388-
"cell_type": "markdown",
389-
"metadata": {},
390-
"source": [
391-
"Note that in a fully-connected feed-forward network, the number of units in each layer always decreases. The neural network is forced to condense information, step-by-step, until it computes the target output we desire. When solving prediction problems, we will rarely (if ever) have a later layer have more neurons than a previous layer."
392-
]
393-
},
394387
{
395388
"cell_type": "markdown",
396389
"metadata": {},

0 commit comments

Comments
 (0)