|
|
@@ -109,10 +109,13 @@ |
|
|
|
"outputs": [], |
|
|
|
"source": [ |
|
|
|
"def vgg_block(num_convs, in_channels, out_channels):\n", |
|
|
|
" net = [nn.Conv2d(in_channels, out_channels, kernel_size=3, padding=1), nn.ReLU(True)] # 定义第一层\n", |
|
|
|
" net = [nn.Conv2d(in_channels, out_channels, \n", |
|
|
|
" kernel_size=3, padding=1), \n", |
|
|
|
" nn.ReLU(True)] # 定义第一层\n", |
|
|
|
"\n", |
|
|
|
" for i in range(num_convs-1): # 定义后面的很多层\n", |
|
|
|
" net.append(nn.Conv2d(out_channels, out_channels, kernel_size=3, padding=1))\n", |
|
|
|
" net.append(nn.Conv2d(out_channels, out_channels, \n", |
|
|
|
" kernel_size=3, padding=1))\n", |
|
|
|
" net.append(nn.ReLU(True))\n", |
|
|
|
" \n", |
|
|
|
" net.append(nn.MaxPool2d(2, 2)) # 定义池化层\n", |
|
|
|