@@ -38,6 +38,7 @@ By default a [`Trainer`] will use the following callbacks:
|
||||
- [`~integrations.CodeCarbonCallback`] if [codecarbon](https://pypi.org/project/codecarbon/) is
|
||||
installed.
|
||||
- [`~integrations.ClearMLCallback`] if [clearml](https://github.com/allegroai/clearml) is installed.
|
||||
- [`~integrations.DagsHubCallback`] if [dagshub](https://dagshub.com/) is installed.
|
||||
|
||||
The main class that implements callbacks is [`TrainerCallback`]. It gets the
|
||||
[`TrainingArguments`] used to instantiate the [`Trainer`], can access that
|
||||
@@ -76,6 +77,8 @@ Here is the list of the available [`TrainerCallback`] in the library:
|
||||
|
||||
[[autodoc]] integrations.ClearMLCallback
|
||||
|
||||
[[autodoc]] integrations.DagsHubCallback
|
||||
|
||||
## TrainerCallback
|
||||
|
||||
[[autodoc]] TrainerCallback
|
||||
|
||||
@@ -1053,7 +1053,7 @@ class MLflowCallback(TrainerCallback):
|
||||
|
||||
class DagsHubCallback(MLflowCallback):
|
||||
"""
|
||||
A [`TrainerCallback`] that logs to [DagsHub](https://dagshub.com/).
|
||||
A [`TrainerCallback`] that logs to [DagsHub](https://dagshub.com/). Extends [`MLflowCallback`]
|
||||
"""
|
||||
|
||||
def __init__(self):
|
||||
@@ -1070,7 +1070,7 @@ class DagsHubCallback(MLflowCallback):
|
||||
Setup the DagsHub's Logging integration.
|
||||
|
||||
Environment:
|
||||
HF_DAGSHUB_LOG_ARTIFACTS (`str`, *optional*):
|
||||
- **HF_DAGSHUB_LOG_ARTIFACTS** (`str`, *optional*):
|
||||
Whether to save the data and model artifacts for the experiment. Default to `False`.
|
||||
"""
|
||||
|
||||
|
||||
Reference in New Issue
Block a user