densenet201 / README.md
Abhirath Anand
Update README.md
4a39454
|
raw
history blame
544 Bytes
metadata
license: mit

DenseNet201 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(201; pretrain = true)