xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-05-24 14:44:21 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

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

Browse Source 
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
John-Henry Lim 2023-05-28 22:36:47 +08:00 committed by GitHub
parent 0c62f85750
commit 1a26e1e195
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ultralytics/yolo/engine/model.py
View File

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