|
|
@ -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'") |
|
|
|