Profiling

Per-layer profiling for deep analysis of model performance

---

source

LayerProfiler

def LayerProfiler(
    model:nn.Module, # model to profile
    sample:torch.Tensor, # input tensor (with batch dimension)
):

Unified per-layer profiler for multiple metrics (speed, memory, size, compute).