A WandB Water & Carbon Metering Plug-in

A wandb and codecarbon wrapper that logs on cloud the water and carbon consumption of your experiments.

Installation

pip install wandc

Usage

Basic usage

from wandc.emissions_logger import log_emissions

@log_emissions()
def your_function_to_track(project_name='your project name'):
  # your code

For other useful parameters see codecarbon documentation here.

Tracking emissions of transformers trainer

from wandc.emissions_logger import log_emissions
from transformers import Trainer

class WandcTrainer(Trainer):
    @log_emissions()
    def train(
        self,
        ...,
        **kwargs,
    ):
        super().train(..., **kwargs)

trainer = WandcTrainer(...)
trainer.train()