Contents

struct

SushiAI::Data::MnistOptions

Where the IDX files are and how much of them to use.

Declared in
include/SushiAI/data/dataset.hpp

Public attributes

std::filesystem::path directory = "data/mnist"

The directory holding the four standard IDX files.

int64_t batch_size = 64

Samples per batch.

int64_t max_samples = 4096

Cap on how many samples to load; 0 means all of them.

bool test_split = false

Load the test split rather than the training split.