Package ai.djl.nn.norm

Class Dropout.Builder

java.lang.Object

ai.djl.nn.norm.Dropout.Builder

Enclosing class:

Dropout

public static final class Dropout.Builder
extends Object

The Builder to construct a Dropout type of Block.

Method Summary

Modifier and Type	Method	Description
Dropout	build()	Builds a Dropout block.
Dropout.Builder	optRate(float rate)	Sets the probability or the fraction of the input that gets dropped out during training time.

Methods inherited from class java.lang.Object

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

Method Details

optRate

public Dropout.Builder optRate(float rate)

Sets the probability or the fraction of the input that gets dropped out during training time. Defaults to 0.5.

Parameters:

rate - fraction of the input that gets dropped out during training

Returns:

this Builder

build

public Dropout build()

Builds a Dropout block.

Returns:

the Dropout block