Spaces:
Running
on
Zero
Running
on
Zero
| import json | |
| from collections import OrderedDict | |
| from dataclasses import dataclass | |
| class EmissionsData: | |
| """ | |
| Output object containing run data | |
| """ | |
| timestamp: str | |
| project_name: str | |
| run_id: str | |
| experiment_id: str | |
| duration: float | |
| emissions: float | |
| emissions_rate: float | |
| cpu_power: float | |
| gpu_power: float | |
| ram_power: float | |
| cpu_energy: float | |
| gpu_energy: float | |
| ram_energy: float | |
| energy_consumed: float | |
| country_name: str | |
| country_iso_code: str | |
| region: str | |
| cloud_provider: str | |
| cloud_region: str | |
| os: str | |
| python_version: str | |
| codecarbon_version: str | |
| cpu_count: float | |
| cpu_model: str | |
| gpu_count: float | |
| gpu_model: str | |
| longitude: float | |
| latitude: float | |
| ram_total_size: float | |
| tracking_mode: str | |
| on_cloud: str = "N" | |
| pue: float = 1 | |
| def values(self) -> OrderedDict: | |
| return OrderedDict(self.__dict__.items()) | |
| def compute_delta_emission(self, previous_emission): | |
| delta_duration = self.duration - previous_emission.duration | |
| self.duration = delta_duration | |
| delta_emissions = self.emissions - previous_emission.emissions | |
| self.emissions = delta_emissions | |
| self.cpu_energy -= previous_emission.cpu_energy | |
| self.gpu_energy -= previous_emission.gpu_energy | |
| self.ram_energy -= previous_emission.ram_energy | |
| self.energy_consumed -= previous_emission.energy_consumed | |
| if delta_duration > 0: | |
| # emissions_rate in g/s : delta_emissions in kg.CO2 / delta_duration in s | |
| self.emissions_rate = delta_emissions / delta_duration | |
| else: | |
| self.emissions_rate = 0 | |
| def toJSON(self): | |
| return json.dumps(self, default=lambda o: o.__dict__, sort_keys=True, indent=4) | |
| class TaskEmissionsData: | |
| task_name: str | |
| timestamp: str | |
| project_name: str | |
| run_id: str | |
| duration: float | |
| emissions: float | |
| emissions_rate: float | |
| cpu_power: float | |
| gpu_power: float | |
| ram_power: float | |
| cpu_energy: float | |
| gpu_energy: float | |
| ram_energy: float | |
| energy_consumed: float | |
| country_name: str | |
| country_iso_code: str | |
| region: str | |
| cloud_provider: str | |
| cloud_region: str | |
| os: str | |
| python_version: str | |
| codecarbon_version: str | |
| cpu_count: float | |
| cpu_model: str | |
| gpu_count: float | |
| gpu_model: str | |
| longitude: float | |
| latitude: float | |
| ram_total_size: float | |
| tracking_mode: str | |
| on_cloud: str = "N" | |
| def values(self) -> OrderedDict: | |
| return OrderedDict(self.__dict__.items()) | |