Package ai.djl.basicmodelzoo.cv.classification

Class LeNet.Builder

java.lang.Object

ai.djl.basicmodelzoo.cv.classification.LeNet.Builder

Enclosing class:

LeNet

public static final class LeNet.Builder
extends Object

The Builder to construct a LeNet object.

Method Summary

Modifier and Type	Method	Description
ai.djl.nn.Block	build()	Builds a LeNet block.
LeNet.Builder	setNumChannels(int[] numChannels)	Sets the number of channels for the LeNet blocks.
LeNet.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

setNumChannels

public LeNet.Builder setNumChannels(int[] numChannels)

Sets the number of channels for the LeNet blocks.

Parameters:

numChannels - the number of channels for every LeNet block.

Returns:

this Builder

setOutSize

public LeNet.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 LeNet block.

Returns:

the LeNet block