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

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
pull/2719/head
John-Henry Lim 2 years ago committed by GitHub
parent 0c62f85750
commit 1a26e1e195
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 2
      ultralytics/yolo/engine/model.py

@ -353,10 +353,10 @@ class YOLO:
kwargs = self.session.train_args kwargs = self.session.train_args
check_pip_update_available() check_pip_update_available()
overrides = self.overrides.copy() overrides = self.overrides.copy()
overrides.update(kwargs)
if kwargs.get('cfg'): if kwargs.get('cfg'):
LOGGER.info(f"cfg file passed. Overriding default params with {kwargs['cfg']}.") LOGGER.info(f"cfg file passed. Overriding default params with {kwargs['cfg']}.")
overrides = yaml_load(check_yaml(kwargs['cfg'])) overrides = yaml_load(check_yaml(kwargs['cfg']))
overrides.update(kwargs)
overrides['mode'] = 'train' overrides['mode'] = 'train'
if not overrides.get('data'): if not overrides.get('data'):
raise AttributeError("Dataset required but missing, i.e. pass 'data=coco128.yaml'") raise AttributeError("Dataset required but missing, i.e. pass 'data=coco128.yaml'")

Loading…
Cancel
Save