| Modifier and Type | Method and Description |
|---|---|
ai.djl.nn.Block |
build()
Builds a
LeNet block. |
LeNet.Builder |
setNumChannels(int[] numChannels)
Sets the number of channels for the LeNet blocks.
|
public LeNet.Builder setNumChannels(int[] numChannels)
numChannels - the number of channels for every LeNet block.Builder