kwargs override cfg argument in model.train (#2839) (#2853)

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-05-28 16:36:47 +02:00
parent 0c62f85750
commit 1a26e1e195
1 changed files with 1 additions and 1 deletions
@@ -353,10 +353,10 @@ class YOLO:
            kwargs = self.session.train_args
        check_pip_update_available()
        overrides = self.overrides.copy()
-        overrides.update(kwargs)
        if kwargs.get('cfg'):
            LOGGER.info(f"cfg file passed. Overriding default params with {kwargs['cfg']}.")
            overrides = yaml_load(check_yaml(kwargs['cfg']))
+        overrides.update(kwargs)
        overrides['mode'] = 'train'
        if not overrides.get('data'):
            raise AttributeError("Dataset required but missing, i.e. pass 'data=coco128.yaml'")