Class AlexNet.Builder
- java.lang.Object
-
- ai.djl.basicmodelzoo.cv.classification.AlexNet.Builder
-
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description ai.djl.nn.Block
build()
Builds aAlexNet
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.AlexNet.Builder
setOutSize(long outSize)
Sets the size of the output.
-
-
-
Method Detail
-
setDropOutRate
public AlexNet.Builder setDropOutRate(float dropOutRate)
Sets the dropout rate in the network.- Parameters:
dropOutRate
- the dropout rate- Returns:
- this
Builder
-
setNumChannels
public AlexNet.Builder setNumChannels(int[] numChannels)
Sets the number of channels for the AlexNet blocks.- Parameters:
numChannels
- the number of channels for every AlexNet block.- Returns:
- this
Builder
-
setOutSize
public AlexNet.Builder setOutSize(long outSize)
Sets the size of the output.- Parameters:
outSize
- the output size- Returns:
- this
Builder
-
-