time_timeable.py

Contents

time_timeable.py#

Script to time a timeable action.

The nip package contains a number of ‘timeable’ actions, such as running a model with a certain input. These actions can be timed to measure their performance. This script provides a command-line interface to time these actions.

Run the script with the --help flag to see all available arguments.

scripts/time_timeable.py#

Time a timeable action

usage: scripts/time_timeable.py [-h] [-a] [-l] [--param-scale PARAM_SCALE]
                                [--no-record-shapes] [--no-profile-memory]
                                [--with-stack]
                                [NAME ...]

name#: The names of the timeables to time

-h, --help#: show this help message and exit

-a, --all#: Time all available timeables

-l, --list#: List all available timeables

--param-scale <param_scale>#: The scale factor for key default parameters

--no-record-shapes#: Don’t record tensor shapes

--no-profile-memory#: Don’t profile memory usage

--with-stack#: Record the stack trace. This introduces a significant additional overhead