diff --git a/Project.toml b/Project.toml index 3e7d07f4..d865a5d0 100644 --- a/Project.toml +++ b/Project.toml @@ -1,6 +1,6 @@ name = "Metalhead" uuid = "dbeba491-748d-5e0e-a39e-b530a07fa0cc" -version = "0.9.2" +version = "0.9.3" [deps] Artifacts = "56f22d72-fd6d-98f1-02f0-08ddc0907c33" diff --git a/src/convnets/vgg.jl b/src/convnets/vgg.jl index ea9f42fe..46876c6f 100644 --- a/src/convnets/vgg.jl +++ b/src/convnets/vgg.jl @@ -109,7 +109,7 @@ Create a VGG style model with specified `depth`. ([reference](https://arxiv.org/abs/1409.1556v6)). !!! warning - + `VGG` does not currently support pretrained weights for the `batchnorm = true` option. # Arguments @@ -130,7 +130,8 @@ end function VGG(depth::Integer; pretrain::Bool = false, batchnorm::Bool = false, inchannels::Integer = 3, nclasses::Integer = 1000) _checkconfig(depth, keys(VGG_CONFIGS)) - layers = vgg((224, 224); config = VGG_CONFIGS[depth], batchnorm, inchannels, nclasses) + layers = vgg((224, 224); config = VGG_CONFIGS[depth], batchnorm, inchannels, nclasses, + dropout_prob = 0.5) model = VGG(layers) if pretrain artifact_name = string("vgg", depth)