From db3c0400c5d59376a5ba5fe1c6c02ecdff8329e3 Mon Sep 17 00:00:00 2001 From: Laughing <61612323+Laughing-q@users.noreply.github.com> Date: Wed, 2 Oct 2024 19:40:47 +0800 Subject: [PATCH] Fix overridden train_args when close_mosaic (#16627) Co-authored-by: UltralyticsAssistant Co-authored-by: Glenn Jocher --- ultralytics/engine/trainer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ultralytics/engine/trainer.py b/ultralytics/engine/trainer.py index 169f4bdf..9fcc6970 100644 --- a/ultralytics/engine/trainer.py +++ b/ultralytics/engine/trainer.py @@ -12,7 +12,7 @@ import os import subprocess import time import warnings -from copy import deepcopy +from copy import copy, deepcopy from datetime import datetime, timedelta from pathlib import Path @@ -749,7 +749,7 @@ class BaseTrainer: self.train_loader.dataset.mosaic = False if hasattr(self.train_loader.dataset, "close_mosaic"): LOGGER.info("Closing dataloader mosaic") - self.train_loader.dataset.close_mosaic(hyp=self.args) + self.train_loader.dataset.close_mosaic(hyp=copy(self.args)) def build_optimizer(self, model, name="auto", lr=0.001, momentum=0.9, decay=1e-5, iterations=1e5): """