ai.djl.basicmodelzoo.cv.classification

Class AlexNet.Builder

java.lang.Object

ai.djl.basicmodelzoo.cv.classification.AlexNet.Builder

Enclosing class:

AlexNet

public static final class AlexNet.Builder
extends java.lang.Object

The Builder to construct a AlexNet object.

Method Summary

Modifier and Type	Method and Description
ai.djl.nn.Block	build() Builds a AlexNet block.
AlexNet.Builder	setDropOutRate(float dropOutRate) Sets the dropout rate in the network.
AlexNet.Builder	setNumChannels(int[] numChannels) Sets the number of channels for the AlexNet blocks.

Methods inherited from class java.lang.Object

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

Method Detail

setDropOutRate

public AlexNet.Builder setDropOutRate(float dropOutRate)

Sets the dropout rate in the network.

Parameters:

dropOutRate - the dropout rate

Returns:

this Builder

setNumChannels

public AlexNet.Builder setNumChannels(int[] numChannels)

Sets the number of channels for the AlexNet blocks.

Parameters:

numChannels - the number of channels for every AlexNet block.

Returns:

this Builder

build

public ai.djl.nn.Block build()

Builds a AlexNet block.

Returns:

the AlexNet block