composer.profiler.profiler_action#

composer.profiler.profiler_action

Classes

ProfilerAction

Action states for the Profiler that define whether or not events are being recorded to the trace file.
class composer.profiler.profiler_action.ProfilerAction(value)[source]#

Bases: composer.utils.string_enum.StringEnum

Action states for the Profiler that define whether or not events are being recorded to the trace file.

SKIP#

Do not record new events to the trace. Any events started during ACTIVE or WARMUP will be recorded upon finish.

WARMUP#

Record all events to the trace except those requiring a warmup period to initialize data structures (e.g., torch.profiler).

ACTIVE#

Record all events to the trace.

ACTIVE_AND_SAVE#

Record all events and save the trace at the end of the batch.