Update CM20315_Convolution_II.ipynb
This commit is contained in:
@@ -105,7 +105,7 @@
|
||||
"cell_type": "code",
|
||||
"source": [
|
||||
"\n",
|
||||
"# TODO Create a model with the folowing layers\n",
|
||||
"# TODO Create a model with the following layers\n",
|
||||
"# 1. Convolutional layer, (input=length 40 and 1 channel, kernel size 3x3, stride 2, padding=\"valid\", 15 output channels ) \n",
|
||||
"# 2. ReLU\n",
|
||||
"# 3. Convolutional layer, (input=length 19 and 15 channels, kernel size 3x3, stride 2, padding=\"valid\", 15 output channels )\n",
|
||||
@@ -120,7 +120,7 @@
|
||||
"# https://pytorch.org/docs/1.13/generated/torch.nn.Linear.html?highlight=linear#torch.nn.Linear\n",
|
||||
"\n",
|
||||
"# Replace the following function which just runs a standard fully connected network\n",
|
||||
"# The flatten at the beginning is becuase we are passing in the data in a slightly different format.\n",
|
||||
"# The flatten at the beginning is because we are passing in the data in a slightly different format.\n",
|
||||
"model = nn.Sequential(\n",
|
||||
"nn.Flatten(),\n",
|
||||
"nn.Linear(40, 100),\n",
|
||||
|
||||
Reference in New Issue
Block a user