xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-05-23 21:44:22 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

DVCLive cache images (#3459)

Browse Source
This commit is contained in:
CptCaptain 2023-07-03 16:29:40 +02:00 committed by GitHub
parent 5e38c7d71b
commit 0e5a6b8158
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
requirements.txt
View File

@ -14,7 +14,7 @@ tqdm>=4.64.0
# Logging ------------------------------------- # Logging -------------------------------------
# tensorboard>=2.13.0 # tensorboard>=2.13.0
# dvclive>=2.11.0 # dvclive>=2.12.0
# clearml # clearml
# comet # comet

2
ultralytics/yolo/utils/callbacks/dvc.py
View File

@ -67,7 +67,7 @@ def on_pretrain_routine_start(trainer):
try: try:
global live global live
if not _logger_disabled(): if not _logger_disabled():
live = dvclive.Live(save_dvc_exp=True) live = dvclive.Live(save_dvc_exp=True, cache_images=True)
LOGGER.info( LOGGER.info(
'DVCLive is detected and auto logging is enabled (can be disabled with `ULTRALYTICS_DVC_DISABLED=true`).' 'DVCLive is detected and auto logging is enabled (can be disabled with `ULTRALYTICS_DVC_DISABLED=true`).'
) )