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.ObjectThe Builder to construct aMobileNetV1object.
-
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description ai.djl.nn.Blockbuild()Builds aMobileNetV1block.MobileNetV1.BuilderoptBatchNormMomentum(float batchNormMomentum)Sets the momentum of batchNorm layer.MobileNetV1.BuilderoptWidthMultiplier(float widthMultiplier)Sets the widthMultiplier of MobileNet.MobileNetV1.BuildersetOutSize(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 aMobileNetV1block.- Returns:
- the
MobileNetV1block
-
-