DVCLive cache images (#3459)

2 years ago · 0e5a6b8158
parent 5e38c7d71b
commit 0e5a6b8158
2 changed files with 2 additions and 2 deletions
--- a/requirements.txt
+++ b/requirements.txt
@ -14,7 +14,7 @@ tqdm>=4.64.0
 # Logging -------------------------------------
 # tensorboard>=2.13.0
-# dvclive>=2.11.0
+# dvclive>=2.12.0
 # clearml
 # comet
--- 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`).'
            )