Compute

Compute modules for benchmarking

---

source

compute_compute

 compute_compute (model:torch.nn.modules.module.Module,
                  dummy_input:torch.Tensor)

Return ComputeMetrics for a single forward pass.

---

source

ComputeMetrics

 ComputeMetrics (macs_m:float|str, params_m:float)

MACs (in millions) and parameter count (also millions).