densenet121 / README.md
Abhirath Anand
Update README.md
442cd2e
metadata
license: mit

DenseNet121 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 = DenseNet(121; pretrain = true)