Merge pull request #115 from Alexander-Barth/vgg-fixes

VGG: use stride=2 for MaxPool like PyTorch
FluxML · Feb 8, 2022 · cf6b905 · cf6b905
2 parents db006a6 + d9ac738
commit cf6b905
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/convnets/vgg.jl b/src/convnets/vgg.jl
@@ -42,7 +42,7 @@ function vgg_convolutional_layers(config, batchnorm, inchannels)
   ifilters = inchannels
   for c in config
     append!(layers, vgg_block(ifilters, c..., batchnorm))
-    push!(layers, MaxPool((2,2)))
+    push!(layers, MaxPool((2,2), stride=2))
     ifilters, _ = c
   end
   return layers