public ResNet50function

ResNet50(; pretrain = false, nclasses = 1000)

Create a ResNet-50 model (reference). See also Metalhead.ResNet.

Arguments

  • pretrain: set to true to load pre-trained weights for ImageNet
  • nclasses: the number of output classes

Warning

ResNet50 does not currently support pretrained weights.