ai.djl.basicmodelzoo.cv.classification

Class NiN.Builder

java.lang.Object

ai.djl.basicmodelzoo.cv.classification.NiN.Builder

Enclosing class:

NiN

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

The Builder to construct a NiN object.

Method Summary

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

Methods inherited from class java.lang.Object

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

Method Detail

setDropOutRate

public NiN.Builder setDropOutRate(float dropOutRate)

Sets the dropout rate in the network.

Parameters:

dropOutRate - the dropout rate

Returns:

this Builder

setNumChannels

public NiN.Builder setNumChannels(int[] numChannels)

Sets the number of channels for the niN blocks.

Parameters:

numChannels - the number of channels for every niN block.

Returns:

this Builder

build

public ai.djl.nn.Block build()

Builds a NiN block.

Returns:

the NiN block