Fix densenet AdaptiveMaxPool2d

AnirudhDagar · Jan 20, 2022 · f6b1dd0 · f6b1dd0
1 parent 801e499
commit f6b1dd0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/chapter_convolutional-modern/densenet.md b/chapter_convolutional-modern/densenet.md
@@ -340,7 +340,7 @@ net.add(nn.BatchNorm(),
 net = nn.Sequential(
     b1, *blks,
     nn.BatchNorm2d(num_channels), nn.ReLU(),
-    nn.AdaptiveMaxPool2d((1, 1)),
+    nn.AdaptiveAvgPool2d((1, 1)),
     nn.Flatten(),
     nn.Linear(num_channels, 10))
 ```