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 java.lang.Object

The Builder to construct a LeNet object.

Method Summary

Modifier and Type	Method and Description
ai.djl.nn.Block	build() Builds a LeNet block.
LeNet.Builder	setNumChannels(int[] numChannels) Sets the number of channels for the LeNet blocks.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Detail

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

build

public ai.djl.nn.Block build()

Builds a LeNet block.

Returns:

the LeNet block