Adds dvclive callback (#27352)

* dvclive trainer callback

* style fixes

* dvclive link fixes
This commit is contained in:
Dave Berenbaum
2023-11-09 07:19:31 -05:00
committed by GitHub
parent c5d7754b11
commit 791ec370d1
7 changed files with 119 additions and 5 deletions

View File

@@ -45,6 +45,7 @@ rendered properly in your Markdown viewer.
- [`~integrations.ClearMLCallback`] [clearml](https://github.com/allegroai/clearml) がインストールされている場合。
- [`~integrations.DagsHubCallback`] [dagshub](https://dagshub.com/) がインストールされている場合。
- [`~integrations.FlyteCallback`] [flyte](https://flyte.org/) がインストールされている場合。
- [`~integrations.DVCLiveCallback`] [dvclive](https://www.dvc.org/doc/dvclive) がインストールされている場合。
パッケージがインストールされているが、付随する統合を使用したくない場合は、`TrainingArguments.report_to` を、使用したい統合のみのリストに変更できます (例: `["azure_ml", "wandb"]`) 。
@@ -88,6 +89,9 @@ rendered properly in your Markdown viewer.
[[autodoc]] integrations.FlyteCallback
[[autodoc]] integrations.DVCLiveCallback
- setup
## TrainerCallback
[[autodoc]] TrainerCallback