Class AlexNet
- java.lang.Object
-
- ai.djl.basicmodelzoo.cv.classification.AlexNet
-
public final class AlexNet extends java.lang.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 class
AlexNet.Builder
The Builder to construct aAlexNet
object.
-
Method Summary
All Methods Static Methods Concrete Methods 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 aAlexNet
.
-
-
-
Method Detail
-
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
- theAlexNet.Builder
with the necessary arguments.- Returns:
- a AlexNet block.
-
builder
public static AlexNet.Builder builder()
Creates a builder to build aAlexNet
.- Returns:
- a new builder
-
-