profile#

pytorch_forecasting.utils._utils.profile(function: Callable, profile_fname: str, filter: str = '', period=0.0001, **kwargs)[source]#

Profile a given function with vmprof.

Parameters:

  • function (Callable) – Function to profile.

  • profile_fname (str) – Path where to save profile (.txt file will be saved with line profile).

  • filter (str, optional) – Filter name (e.g. module name) to filter profile. Defaults to “”.

  • period (float, optional) – Frequency of calling profiler in seconds. Defaults to 0.0001.