Class NiN.Builder
java.lang.Object
ai.djl.basicmodelzoo.cv.classification.NiN.Builder
- Enclosing class:
- NiN
The Builder to construct a
NiN
object.-
Method Summary
Modifier and TypeMethodDescriptionai.djl.nn.Block
build()
Builds aNiN
block.setDropOutRate
(float dropOutRate) Sets the dropout rate in the network.setNumChannels
(int[] numChannels) Sets the number of channels for the niN blocks.
-
Method Details
-
setDropOutRate
Sets the dropout rate in the network.- Parameters:
dropOutRate
- the dropout rate- Returns:
- this
Builder
-
setNumChannels
Sets the number of channels for the niN blocks.- Parameters:
numChannels
- the number of channels for every niN block.- Returns:
- this
Builder
-
build
public ai.djl.nn.Block build()Builds aNiN
block.- Returns:
- the
NiN
block
-