| Constructor and Description |
|---|
Builder() |
| Modifier and Type | Method and Description |
|---|---|
ai.djl.nn.Block |
build()
Builds a
VGG block. |
VGG.Builder |
setConvArch(int[][] convArch)
Sets the number of blocks according to the user.
|
VGG.Builder |
setNumLayers(int numLayers)
Sets the number of layers.
|
public VGG.Builder setNumLayers(int numLayers)
numLayers - the number of layers in the network. default is 11.Builderpublic VGG.Builder setConvArch(int[][] convArch)
convArch - 2-D array consisting of number of convolutions and the number of
channels.Builder