luz_callback_profile: Profile callback

In mlverse/luz: Higher Level 'API' for 'torch'

Profile callback

Description

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

Usage

luz_callback_profile()

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

Value

A luz_callback

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

See Also

Other luz_callbacks: luz_callback_auto_resume(), luz_callback_csv_logger(), luz_callback_early_stopping(), luz_callback_interrupt(), luz_callback_keep_best_model(), luz_callback_lr_scheduler(), luz_callback_metrics(), luz_callback_mixed_precision(), luz_callback_mixup(), luz_callback_model_checkpoint(), luz_callback_progress(), luz_callback_resume_from_checkpoint(), luz_callback_train_valid(), luz_callback()

Examples

profile_callback <- luz_callback_profile()

mlverse/luz documentation built on Sept. 19, 2024, 11:20 p.m.