Package ai.djl.nn.norm

Class Dropout.Builder

  • Enclosing class:
    Dropout
    public static final class Dropout.Builder
extends java.lang.Object
    The Builder to construct a Dropout type of Block.

    • Method Summary

      All Methods Instance Methods Concrete Methods 
      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 Detail

      • 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