Contents

struct

SushiAI::Config::TrainSpec

One training run, end to end.

Declared in
include/SushiAI/config/train_config.hpp

Public attributes

std::filesystem::path model_path

The architecture file, resolved against this file's directory.

std::string objective_type

The Train::OBJECTIVE_REGISTRY key, e.g.

"cross_entropy".

Value objective_attributes

The objective's own keys.

DataSpec data

Where the batches come from.

std::string optimizer_type

The Optim::OPTIMIZER_REGISTRY key, e.g.

"adamw".

Value optimizer_attributes

The optimizer's own keys.

std::int64_t epochs = 1

How many passes over the dataset; at least one.

std::filesystem::path checkpoint

Where to write the checkpoint; empty means do not write one.