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

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()