Update CM20315_Convolution_II.ipynb

This commit is contained in:
Pietro Monticone
2023-11-30 16:33:04 +01:00
parent 4fc1abc20e
commit e03fad482b

View File

@@ -105,7 +105,7 @@
"cell_type": "code", "cell_type": "code",
"source": [ "source": [
"\n", "\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", "# 1. Convolutional layer, (input=length 40 and 1 channel, kernel size 3x3, stride 2, padding=\"valid\", 15 output channels ) \n",
"# 2. ReLU\n", "# 2. ReLU\n",
"# 3. Convolutional layer, (input=length 19 and 15 channels, kernel size 3x3, stride 2, padding=\"valid\", 15 output channels )\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", "# https://pytorch.org/docs/1.13/generated/torch.nn.Linear.html?highlight=linear#torch.nn.Linear\n",
"\n", "\n",
"# Replace the following function which just runs a standard fully connected network\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", "model = nn.Sequential(\n",
"nn.Flatten(),\n", "nn.Flatten(),\n",
"nn.Linear(40, 100),\n", "nn.Linear(40, 100),\n",