Skip to content

GitLab

Unverified Commit fcb4f11c authored by Nathan Raw's avatar Nathan Raw Committed by GitHub
Browse files

📝 Add codecarbon callback to docs (#15563)

parent 077c00c0
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
docs/source/main_classes/callback.mdx
View file @ fcb4f11c
......@@ -34,6 +34,8 @@ By default a [`Trainer`] will use the following callbacks:
- [`~integrations.MLflowCallback`] if [mlflow](https://www.mlflow.org/) is installed.
- [`~integrations.AzureMLCallback`] if [azureml-sdk](https://pypi.org/project/azureml-sdk/) is
installed.
- [`~integrations.CodeCarbonCallback`] if [codecarbon](https://pypi.org/project/codecarbon/) is
installed.
The main class that implements callbacks is [`TrainerCallback`]. It gets the
[`TrainingArguments`] used to instantiate the [`Trainer`], can access that
......@@ -66,6 +68,8 @@ Here is the list of the available [`TrainerCallback`] in the library:
[[autodoc]] integrations.AzureMLCallback
[[autodoc]] integrations.CodeCarbonCallback
## TrainerCallback
[[autodoc]] TrainerCallback
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment