Fix overridden train_args when close_mosaic (#16627)

Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
pull/16632/head
Laughing 5 months ago committed by GitHub
parent 3789cde7dd
commit db3c0400c5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
  1. 4
      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):
"""

Loading…
Cancel
Save