Energy

Energy modules for benchmarking

---

source

compute_energy_multi

 compute_energy_multi (model:torch.nn.modules.module.Module,
                       x:torch.Tensor,
                       devices:Optional[Sequence[str|torch.device]]=None,
                       **kw)

---

source

compute_energy

 compute_energy (model:torch.nn.modules.module.Module, x:torch.Tensor,
                 device:str|torch.device='cuda', warmup:int=20,
                 steps:int=100, offline:bool=True,
                 country_iso:str|None=None, measure_secs:int=1)

---

source

EnergyMetrics

 EnergyMetrics (mean_watts:float, energy_wh:float, co2_eq_g:float)