ai.djl.basicdataset

Class Mnist.Builder

java.lang.Object

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

ai.djl.basicdataset.Mnist.Builder

Enclosing class:

Mnist

public static final class Mnist.Builder
extends ai.djl.training.dataset.RandomAccessDataset.BaseBuilder<Mnist.Builder>

A builder for a Mnist.

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
Mnist	build() Builds the Mnist.
Mnist.Builder	optArtifact(ai.djl.repository.Artifact artifact) Sets the optional artifact.
Mnist.Builder	optRepository(ai.djl.repository.Repository repository) Sets the optional repository.
Mnist.Builder	optUsage(ai.djl.training.dataset.Dataset.Usage usage) Sets the optional usage.
protected Mnist.Builder	self()
Mnist.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 Mnist.Builder self()

Specified by:

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

setManager

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

Sets the manager for the dataset.

Parameters:

manager - the manager

Returns:

this builder

optRepository

public Mnist.Builder optRepository(ai.djl.repository.Repository repository)

Sets the optional repository.

Parameters:

repository - the repository

Returns:

this builder

optArtifact

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

Sets the optional artifact.

Parameters:

artifact - the artifact

Returns:

this builder

optUsage

public Mnist.Builder optUsage(ai.djl.training.dataset.Dataset.Usage usage)

Sets the optional usage.

Parameters:

usage - the usage

Returns:

this builder

build

public Mnist build()

Builds the Mnist.

Returns:

the Mnist