ai.djl.basicmodelzoo.cv.classification

Class ResNetV1.Builder

java.lang.Object

ai.djl.basicmodelzoo.cv.classification.ResNetV1.Builder

Enclosing class:

ResNetV1

public static final class ResNetV1.Builder
extends java.lang.Object

The Builder to construct a ResNetV1 object.

Method Summary

Modifier and Type	Method and Description
ai.djl.nn.Block	build() Builds a ResNetV1 block.
ResNetV1.Builder	optBatchNormMomentum(float batchNormMomentum) Sets the momentum of batchNorm layer.
ResNetV1.Builder	setImageShape(ai.djl.ndarray.types.Shape imageShape) Sets the shape of the image.
ResNetV1.Builder	setNumLayers(int numLayers) Sets the number of layers in the network.
ResNetV1.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 Detail

setNumLayers

public ResNetV1.Builder setNumLayers(int numLayers)

Sets the number of layers in the network.

Parameters:

numLayers - the number of layers

Returns:

this Builder

setOutSize

public ResNetV1.Builder setOutSize(long outSize)

Sets the size of the output.

Parameters:

outSize - the output size

Returns:

this Builder

optBatchNormMomentum

public ResNetV1.Builder optBatchNormMomentum(float batchNormMomentum)

Sets the momentum of batchNorm layer.

Parameters:

batchNormMomentum - the momentum

Returns:

this Builder

setImageShape

public ResNetV1.Builder setImageShape(ai.djl.ndarray.types.Shape imageShape)

Sets the shape of the image.

Parameters:

imageShape - the shape of the image

Returns:

this Builder

build

public ai.djl.nn.Block build()

Builds a ResNetV1 block.

Returns:

the ResNetV1 block