Class MobileNetV1.Builder
java.lang.Object
ai.djl.basicmodelzoo.cv.classification.MobileNetV1.Builder
- Enclosing class:
- MobileNetV1
The Builder to construct a
MobileNetV1
object.-
Method Summary
Modifier and TypeMethodDescriptionai.djl.nn.Block
build()
Builds aMobileNetV1
block.optBatchNormMomentum
(float batchNormMomentum) Sets the momentum of batchNorm layer.optWidthMultiplier
(float widthMultiplier) Sets the widthMultiplier of MobileNet.setOutSize
(long outSize) Sets the size of the output.
-
Method Details
-
optWidthMultiplier
Sets the widthMultiplier of MobileNet.- Parameters:
widthMultiplier
- the widthMultiplier of MobileNet- Returns:
- this
Builder
-
optBatchNormMomentum
Sets the momentum of batchNorm layer.- Parameters:
batchNormMomentum
- the momentum- Returns:
- this
Builder
-
setOutSize
Sets the size of the output.- Parameters:
outSize
- the output size- Returns:
- this
Builder
-
build
public ai.djl.nn.Block build()Builds aMobileNetV1
block.- Returns:
- the
MobileNetV1
block
-