diff --git a/chapter_convolutional-modern/densenet.md b/chapter_convolutional-modern/densenet.md index 0f526aca7b..5dee70b397 100644 --- 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)) ```