AlexNet model architecture from the One weird trick... paper.

model_alexnet(pretrained = FALSE, progress = TRUE, ...)

Arguments

pretrained

(bool): If TRUE, returns a model pre-trained on ImageNet.

progress

(bool): If TRUE, displays a progress bar of the download to stderr.

...

other parameters passed to the model intializer. currently only num_classes is used.

See also