Add DVC experiments logger with DVCLive (#2792)

This commit is contained in:
Ivan Shcheklein
2023-06-05 05:10:43 -10:00
committed by GitHub
parent 2b26572e42
commit 6057b267af
3 changed files with 138 additions and 1 deletions

View File

@ -198,6 +198,7 @@ def add_integration_callbacks(instance):
"""
from .clearml import callbacks as clearml_cb
from .comet import callbacks as comet_cb
from .dvc import callbacks as dvc_cb
from .hub import callbacks as hub_cb
from .mlflow import callbacks as mlflow_cb
from .neptune import callbacks as neptune_cb
@ -205,7 +206,7 @@ def add_integration_callbacks(instance):
from .tensorboard import callbacks as tensorboard_cb
from .wb import callbacks as wb_cb
for x in clearml_cb, comet_cb, hub_cb, mlflow_cb, neptune_cb, tune_cb, tensorboard_cb, wb_cb:
for x in clearml_cb, comet_cb, hub_cb, mlflow_cb, neptune_cb, tune_cb, tensorboard_cb, wb_cb, dvc_cb:
for k, v in x.items():
if v not in instance.callbacks[k]: # prevent duplicate callbacks addition
instance.callbacks[k].append(v) # callback[name].append(func)