Mila::Dnn::TrainingConfig Struct Reference

Configuration for training a model. More...

Public Attributes
size_t	batch_size = 16
	Batch size for training.
float	beta1 = 0.9f
	Beta1 for Adam optimizer.
float	beta2 = 0.999f
	Beta2 for Adam optimizer.
std::string	checkpoint_dir = ""
	Directory to save checkpoints.
size_t	early_stopping = 0
	Stop after N epochs with no improvement (0 = disabled)
size_t	epochs = 10
	Number of epochs to train.
float	epsilon = 1e-8f
	Epsilon for Adam optimizer.
float	learning_rate = 1e-3f
	Learning rate for optimization.
bool	save_best_only = true
	Save only the best model.
size_t	validation_interval = 1
	Validate every N epochs.
bool	verbose = true
	Print training progress.
float	weight_decay = 0.0f
	Weight decay (L2 regularization)

Detailed Description

Configuration for training a model.

Member Data Documentation

batch_size

size_t Mila::Dnn::TrainingConfig::batch_size = 16

Batch size for training.

beta1

float Mila::Dnn::TrainingConfig::beta1 = 0.9f

Beta1 for Adam optimizer.

beta2

float Mila::Dnn::TrainingConfig::beta2 = 0.999f

Beta2 for Adam optimizer.

checkpoint_dir

std::string Mila::Dnn::TrainingConfig::checkpoint_dir = ""

Directory to save checkpoints.

early_stopping

size_t Mila::Dnn::TrainingConfig::early_stopping = 0

Stop after N epochs with no improvement (0 = disabled)

epochs

size_t Mila::Dnn::TrainingConfig::epochs = 10

Number of epochs to train.

epsilon

float Mila::Dnn::TrainingConfig::epsilon = 1e-8f

Epsilon for Adam optimizer.

learning_rate

float Mila::Dnn::TrainingConfig::learning_rate = 1e-3f

Learning rate for optimization.

save_best_only

bool Mila::Dnn::TrainingConfig::save_best_only = true

Save only the best model.

validation_interval

size_t Mila::Dnn::TrainingConfig::validation_interval = 1

Validate every N epochs.

verbose

bool Mila::Dnn::TrainingConfig::verbose = true

Print training progress.

weight_decay

float Mila::Dnn::TrainingConfig::weight_decay = 0.0f

Weight decay (L2 regularization)

---

The documentation for this struct was generated from the following file:

• /home/runner/work/Mila/Mila/Mila/Src/Dnn/Models/Model.ixx