DVCLive cache images (#3459)

2023-07-03 16:29:40 +02:00
parent 5e38c7d71b
commit 0e5a6b8158
2 changed files with 2 additions and 2 deletions
--- a/ultralytics/yolo/utils/callbacks/dvc.py
+++ b/ultralytics/yolo/utils/callbacks/dvc.py
@ -67,7 +67,7 @@ def on_pretrain_routine_start(trainer):
    try:
        global live
        if not _logger_disabled():
-            live = dvclive.Live(save_dvc_exp=True)
+            live = dvclive.Live(save_dvc_exp=True, cache_images=True)
            LOGGER.info(
                'DVCLive is detected and auto logging is enabled (can be disabled with `ULTRALYTICS_DVC_DISABLED=true`).'
            )