profiler_action

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

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)

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.