vgg11 / README.md
darsnack's picture
Initial README
22bce9d
metadata
license: mit

VGG11 (w/o batch norm) model ported from torchvision for use with Metalhead.jl. The scripts for creating this file can be found at this gist.

To use this model in Julia, add the Metalhead.jl package to your environment. Then execute:

using Metalhead

model = VGG(11; pretrain = true)