pytorch_pfn_extras.profiler.Tracer

class pytorch_pfn_extras.profiler.Tracer

Bases: object

Methods

__init__()
add_event(name)
add_remote_event(name, value)
clear()
enable(enable_flag)
finalize()
flush(filename, writer)
initialize_writer(filename, writer)
load_state_dict(to_load)
state_dict()

add_event(name)

Parameters:

name (str) –

Return type:

Generator[None, None, None]

add_remote_event(name, value)

Parameters:

• name (str) –

• value (Any) –

Return type:

None

clear()

Return type:

None

enable(enable_flag)

Parameters:

enable_flag (bool) –

Return type:

None

finalize()

Return type:

None

flush(filename, writer)

Parameters:

• filename (str) –

• writer (Writer) –

Return type:

None

initialize_writer(filename, writer)

Parameters:

• filename (str) –

• writer (Writer) –

Return type:

None

load_state_dict(to_load)

Parameters:

to_load (Dict[str, Any]) –

Return type:

None

state_dict()

Return type:

Dict[str, Any]