Package ai.djl.basicmodelzoo.cv.classification

Class NiN.Builder

  • Enclosing class:
    NiN
    public static final class NiN.Builder
extends java.lang.Object
    The Builder to construct a NiN object.

    • Method Summary

      All Methods Instance Methods Concrete Methods 
      Modifier and Type Method 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