ai.djl.basicdataset

Class Cifar10.Builder

java.lang.Object

ai.djl.training.dataset.RandomAccessDataset.BaseBuilder<Cifar10.Builder>

ai.djl.basicdataset.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

batchifier, device, executor, maxIteration, pipeline, prefetchNumber, sampler, targetPipeline

Constructor Summary

Constructors

Constructor and Description
Builder() Constructs a new builder.

Method Summary

Modifier and Type	Method and Description
Cifar10	build() Builds a new Cifar10.
Cifar10.Builder	optArtifact(ai.djl.repository.Artifact artifact) Sets the optional artifact containing the data.
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()
Cifar10.Builder	setManager(ai.djl.ndarray.NDManager manager) Sets the manager for the dataset.

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

getSampler, optBatchier, optDevice, optExcutor, optMaxIteration, optPipeline, optTargetPipeline, setSampling, setSampling, setSampling

Methods inherited from class java.lang.Object

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

Constructor Detail

Builder

public Builder()

Constructs a new builder.

Method Detail

self

protected Cifar10.Builder self()

Specified by:

self in class ai.djl.training.dataset.RandomAccessDataset.BaseBuilder<Cifar10.Builder>

setManager

public Cifar10.Builder setManager(ai.djl.ndarray.NDManager manager)

Sets the manager for the dataset.

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

optArtifact

public Cifar10.Builder optArtifact(ai.djl.repository.Artifact artifact)

Sets the optional artifact containing the data.

Parameters:

artifact - the artifact

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

build

public Cifar10 build()

Builds a new Cifar10.

Returns:

the new Cifar10