Package ai.djl.basicmodelzoo.cv.classification

Class AlexNet

java.lang.Object

ai.djl.basicmodelzoo.cv.classification.AlexNet

public final class AlexNet
extends Object

AlexNet contains a generic implementation of AlexNet adapted from [torchvision implmentation](https://github.com/pytorch/vision/blob/master/torchvision/models/alexnet.py)

AlexNet model from the "One weird trick..." https://arxiv.org/abs/1404.5997 paper.

See Also:

• The D2L chapter on AlexNet

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static final class	AlexNet.Builder	The Builder to construct a AlexNet object.

Method Summary

Modifier and Type	Method	Description
static ai.djl.nn.Block	alexNet(AlexNet.Builder builder)	Creates a AlexNet network block with the help of the AlexNet Builder.
static AlexNet.Builder	builder()	Creates a builder to build a AlexNet.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Details

alexNet

public static ai.djl.nn.Block alexNet(AlexNet.Builder builder)

Creates a AlexNet network block with the help of the AlexNet Builder.

Parameters:

builder - the AlexNet.Builder with the necessary arguments.

Returns:

a AlexNet block.

builder

public static AlexNet.Builder builder()

Creates a builder to build a AlexNet.

Returns:

a new builder