Class MobileNetV1.Builder
- java.lang.Object
-
- ai.djl.basicmodelzoo.cv.classification.MobileNetV1.Builder
-
- Enclosing class:
- MobileNetV1
public static final class MobileNetV1.Builder extends java.lang.Object
The Builder to construct aMobileNetV1
object.
-
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description ai.djl.nn.Block
build()
Builds aMobileNetV1
block.MobileNetV1.Builder
optBatchNormMomentum(float batchNormMomentum)
Sets the momentum of batchNorm layer.MobileNetV1.Builder
optWidthMultiplier(float widthMultiplier)
Sets the widthMultiplier of MobileNet.MobileNetV1.Builder
setOutSize(long outSize)
Sets the size of the output.
-
-
-
Method Detail
-
optWidthMultiplier
public MobileNetV1.Builder optWidthMultiplier(float widthMultiplier)
Sets the widthMultiplier of MobileNet.- Parameters:
widthMultiplier
- the widthMultiplier of MobileNet- Returns:
- this
Builder
-
optBatchNormMomentum
public MobileNetV1.Builder optBatchNormMomentum(float batchNormMomentum)
Sets the momentum of batchNorm layer.- Parameters:
batchNormMomentum
- the momentum- Returns:
- this
Builder
-
setOutSize
public MobileNetV1.Builder setOutSize(long outSize)
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
-
-