Package ai.djl.basicmodelzoo.cv.classification

Class MobileNetV1.Builder

java.lang.Object

ai.djl.basicmodelzoo.cv.classification.MobileNetV1.Builder

Enclosing class:

MobileNetV1

public static final class MobileNetV1.Builder
extends Object

The Builder to construct a MobileNetV1 object.

Method Summary

Modifier and Type	Method	Description
ai.djl.nn.Block	build()	Builds a MobileNetV1 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.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Details

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 a MobileNetV1 block.

Returns:

the MobileNetV1 block