public static final class Mnist.Builder extends ai.djl.training.dataset.RandomAccessDataset.BaseBuilder<Mnist.Builder>
Mnist.| 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 |
optManager(ai.djl.ndarray.NDManager manager)
Sets the optional manager for the dataset (default follows engine default).
|
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() |
protected Mnist.Builder self()
self in class ai.djl.training.dataset.RandomAccessDataset.BaseBuilder<Mnist.Builder>public Mnist.Builder optManager(ai.djl.ndarray.NDManager manager)
manager - the managerpublic Mnist.Builder optRepository(ai.djl.repository.Repository repository)
repository - the repositorypublic Mnist.Builder optArtifact(ai.djl.repository.Artifact artifact)
artifact - the artifactpublic Mnist.Builder optUsage(ai.djl.training.dataset.Dataset.Usage usage)
usage - the usage