Skip to contents

Computes the times for high-level operations in the training loops.

Usage

luz_callback_profile()

Value

A luz_callback

Details

Records are saved in ctx$records$profile. Times are stored as seconds. Data is stored in the following structure:

  • fit time for the entire fit procedure.

  • epoch times per epoch

  • (train/valid)_batch time per batch of data processed, including data acquisition and step.

  • (train/valid)_step time per step (training or validation step) - only the model step. (not including data acquisition and preprocessing)

Note

In general you don't need to use these callback by yourself because it's always included by default in fit.luz_module_generator().

Examples

profile_callback <- luz_callback_profile()