Introduction

Tutorials

Developer guide

API Reference

public AlexNetstruct

AlexNet(; pretrain = false, nclasses = 1000)

Create a AlexNet. See also alexnet.

Warning

AlexNet does not currently support pretrained weights.

Arguments

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