Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
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'")
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue