Class ResNetV1.Builder
- java.lang.Object
-
- ai.djl.basicmodelzoo.cv.classification.ResNetV1.Builder
-
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description ai.djl.nn.Block
build()
Builds aResNetV1
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.
-
-
-
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
-
-