nip.timing.timeables.time_all_timeables

nip.timing.timeables.time_all_timeables#

nip.timing.timeables.time_all_timeables(print_results: bool = True, common_kwargs: dict[str, Any] = {}, per_timeable_kwargs: dict[str, dict[str, Any]] = {}) → dict[str, EventList][source]#

Time all timeables.

Parameters:

  • print_results (bool, default=True) – Whether to print the results of the timing.

  • common_kwargs (dict[str, Any], default={}) – A dictionary of keyword arguments that are passed to all timeables.

  • timeable_kwargs (dict[str, dict[str, Any]], default={}) – A dictionary of dictionaries of keyword arguments that are passed to specific timeables, with the timeable name as the key.

Returns:

results (dict[str, torch.autograd.profiler.EventList]) – The profiling results.