Package ai.djl.basicdataset.cv.classification

Class Cifar10.Builder

  • Enclosing class:
    Cifar10
    public static final class Cifar10.Builder
extends ai.djl.training.dataset.RandomAccessDataset.BaseBuilder<Cifar10.Builder>
    A builder to construct a Cifar10.

    • Field Summary

      • Fields inherited from class ai.djl.training.dataset.RandomAccessDataset.BaseBuilder

        dataBatchifier, device, labelBatchifier, limit, pipeline, prefetchNumber, sampler, targetPipeline

    • Method Summary

      All Methods Instance Methods Concrete Methods 
      Modifier and Type Method Description
      Cifar10 build()
      Builds a new Cifar10.
      Cifar10.Builder optArtifactId​(java.lang.String artifactId)
      Sets the optional artifactId.
      Cifar10.Builder optGroupId​(java.lang.String groupId)
      Sets optional groupId.
      Cifar10.Builder optManager​(ai.djl.ndarray.NDManager manager)
      Sets the optional manager for the dataset (default follows engine default).
      Cifar10.Builder optRepository​(ai.djl.repository.Repository repository)
      Sets the optional repository for the dataset.
      Cifar10.Builder optUsage​(ai.djl.training.dataset.Dataset.Usage usage)
      Sets the optional usage for the dataset.
      protected Cifar10.Builder self()

      • Methods inherited from class ai.djl.training.dataset.RandomAccessDataset.BaseBuilder

        addTargetTransform, addTransform, getSampler, optDataBatchifier, optDevice, optLabelBatchifier, optLimit, optPipeline, optPrefetchNumber, optTargetPipeline, setSampling, setSampling, setSampling

      • Methods inherited from class java.lang.Object

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

    • Method Detail

      • optManager

        public Cifar10.Builder optManager​(ai.djl.ndarray.NDManager manager)
        Sets the optional manager for the dataset (default follows engine default).
        Parameters:
        manager - the new manager
        Returns:
        this builder

      • optRepository

        public Cifar10.Builder optRepository​(ai.djl.repository.Repository repository)
        Sets the optional repository for the dataset.
        Parameters:
        repository - the new repository
        Returns:
        this builder

      • optGroupId

        public Cifar10.Builder optGroupId​(java.lang.String groupId)
        Sets optional groupId.
        Parameters:
        groupId - the groupId}
        Returns:
        this builder

      • optArtifactId

        public Cifar10.Builder optArtifactId​(java.lang.String artifactId)
        Sets the optional artifactId.
        Parameters:
        artifactId - the artifactId
        Returns:
        this builder

      • optUsage

        public Cifar10.Builder optUsage​(ai.djl.training.dataset.Dataset.Usage usage)
        Sets the optional usage for the dataset.
        Parameters:
        usage - the usage
        Returns:
        this builder