ResNetV1.Builder (Deep Java Library 0.16.0 - model-zoo API)

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

All Methods Instance Methods Concrete Methods
Modifier and Type	Method	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