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`).' )