From 0e5a6b8158dd851ea8bedf8080cea9e07278714e Mon Sep 17 00:00:00 2001
From: CptCaptain <nilskoch@posteo.de>
Date: Mon, 3 Jul 2023 16:29:40 +0200
Subject: [PATCH] DVCLive cache images (#3459)

---
 requirements.txt                        | 2 +-
 ultralytics/yolo/utils/callbacks/dvc.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index c062769..be1fce9 100644
--- 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
 
diff --git a/ultralytics/yolo/utils/callbacks/dvc.py b/ultralytics/yolo/utils/callbacks/dvc.py
index 63ec368..138100c 100644
--- 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`).'
             )