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.
| Modifier and Type | Class and Description |
|---|---|
static class |
AlexNet.Builder
The Builder to construct a
AlexNet object. |
| Modifier and Type | Method and 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. |
public static ai.djl.nn.Block alexNet(AlexNet.Builder builder)
builder - the AlexNet.Builder with the necessary arguments.public static AlexNet.Builder builder()
AlexNet.