pyg_spectral.profile

class Stopwatch[source]

Bases: object

start()[source]
pause() → float[source]

Pause clocking and return elapsed time

lap() → float[source]

No pausing, return elapsed time

reset()[source]
property data: float[source]
class Accumulator[source]

Bases: object

reset()[source]
update(val: float, count: int = 1)[source]
property mean: float[source]
class MemoryRAM[source]

Bases: NumFmt

Memory usage of current process in RAM.

update()[source]
class MemoryCUDA[source]

Bases: NumFmt

Memory usage of current process in CUDA.

update()[source]
class ParamNumel(module: Module | None = None)[source]

Bases: NumFmt

Number of learnable parameters in an torch.nn.Module.

update(module: Module)[source]
class ParamMemory(module: Module | None = None)[source]

Bases: NumFmt

Memory usage of parameters in an torch.nn.Module.

update(module: Module)[source]