mnist_dataset: MNIST dataset

In mlverse/torchvision: Models, Datasets and Transformations for Images

MNIST dataset

Description

Prepares the MNIST dataset and optionally downloads it.

Usage

mnist_dataset(
  root,
  train = TRUE,
  transform = NULL,
  target_transform = NULL,
  download = FALSE
)

Arguments

root	(string): Root directory of dataset where MNIST/processed/training.pt and MNIST/processed/test.pt exist.
train	(bool, optional): If True, creates dataset from training.pt, otherwise from test.pt.
transform	(callable, optional): A function/transform that takes in an PIL image and returns a transformed version. E.g, transform_random_crop().
target_transform	(callable, optional): A function/transform that takes in the target and transforms it.
download	(bool, optional): If true, downloads the dataset from the internet and puts it in root directory. If dataset is already downloaded, it is not downloaded again.

mlverse/torchvision documentation built on Sept. 18, 2024, 4:03 p.m.