max_steps_per_epoch

Description

max_steps_per_epoch controls the maximum number of steps performed in a single epoch during training.

• A step is defined as training your model on a single mini-batch of size batch_size
• An epoch is defined as the process of completing the specified number of steps, i.e., max_steps_per_epoch steps, after which validation is performed.

This is particularly useful for managing the trade-off between training time and validation frequency.

For example, to double the validation frequency while keeping the total number of mini-batches the model processes unchanged, halve max_steps_per_epoch and double max_epochs.

• max_steps_per_epoch must be > 0.

Supported Task Types

• All

Default Values