Package ai.djl.basicmodelzoo.cv.classification

Class LeNet.Builder

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

    • Method Summary

      All Methods Instance Methods Concrete Methods 
      Modifier and Type Method Description
      ai.djl.nn.Block build()
      Builds a LeNet block.
      LeNet.Builder setNumChannels​(int[] numChannels)
      Sets the number of channels for the LeNet blocks.

      • Methods inherited from class java.lang.Object

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

    • Method Detail

      • setNumChannels

        public LeNet.Builder setNumChannels​(int[] numChannels)
        Sets the number of channels for the LeNet blocks.
        Parameters:
        numChannels - the number of channels for every LeNet block.
        Returns:
        this Builder

      • build

        public ai.djl.nn.Block build()
        Builds a LeNet block.
        Returns:
        the LeNet block