Package ai.djl.basicmodelzoo.cv.classification

Class AlexNet

  • public final class AlexNet
extends java.lang.Object
    AlexNet contains a generic implementation of AlexNet adapted from [torchvision implmentation](https://github.com/pytorch/vision/blob/master/torchvision/models/alexnet.py)

    AlexNet model from the "One weird trick..." https://arxiv.org/abs/1404.5997 paper.

    See Also:
    The D2L chapter on AlexNet

    • Nested Class Summary

      Nested Classes 
      Modifier and Type Class Description
      static class  AlexNet.Builder
      The Builder to construct a AlexNet object.

    • Method Summary

      All Methods Static Methods Concrete Methods 
      Modifier and Type Method Description
      static ai.djl.nn.Block alexNet​(AlexNet.Builder builder)
      Creates a AlexNet network block with the help of the AlexNet Builder.
      static AlexNet.Builder builder()
      Creates a builder to build a AlexNet.

      • Methods inherited from class java.lang.Object

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

    • Method Detail

      • alexNet

        public static ai.djl.nn.Block alexNet​(AlexNet.Builder builder)
        Creates a AlexNet network block with the help of the AlexNet Builder.
        Parameters:
        builder - the AlexNet.Builder with the necessary arguments.
        Returns:
        a AlexNet block.

      • builder

        public static AlexNet.Builder builder()
        Creates a builder to build a AlexNet.
        Returns:
        a new builder