| Modifier and Type | Method and Description |
|---|---|
ai.djl.nn.Block |
build()
Builds a
AlexNet block. |
AlexNet.Builder |
setDropOutRate(float dropOutRate)
Sets the dropout rate in the network.
|
AlexNet.Builder |
setNumChannels(int[] numChannels)
Sets the number of channels for the AlexNet blocks.
|
public AlexNet.Builder setDropOutRate(float dropOutRate)
dropOutRate - the dropout rateBuilderpublic AlexNet.Builder setNumChannels(int[] numChannels)
numChannels - the number of channels for every AlexNet block.Builder