Class AlexNet
java.lang.Object
ai.djl.basicmodelzoo.cv.classification.AlexNet
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:
-
Nested Class Summary
-
Method Summary
Modifier and TypeMethodDescriptionstatic 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 aAlexNet
.
-
Method Details
-
alexNet
Creates a AlexNet network block with the help of the AlexNet Builder.- Parameters:
builder
- theAlexNet.Builder
with the necessary arguments.- Returns:
- a AlexNet block.
-
builder
Creates a builder to build aAlexNet
.- Returns:
- a new builder
-