resnet50 / README.md
darsnack's picture
Create README.md
85406ec
|
raw
history blame
539 Bytes
metadata
license: mit

ResNet50 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 = ResNet(50; pretrain = true)