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

Fix incorrect device selection when resuming training (#4301)

Browse Source
This commit is contained in:
Laughing 2023-08-11 20:32:39 +08:00 committed by GitHub
parent 28893d9d2d
commit b73f9b13df
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/engine/trainer.py
View File

@ -80,8 +80,8 @@ class BaseTrainer:
overrides (dict, optional): Configuration overrides. Defaults to None. overrides (dict, optional): Configuration overrides. Defaults to None.
""" """
self.args = get_cfg(cfg, overrides) self.args = get_cfg(cfg, overrides)
self.device = select_device(self.args.device, self.args.batch)
self.check_resume() self.check_resume()
self.device = select_device(self.args.device, self.args.batch)
self.validator = None self.validator = None
self.model = None self.model = None
self.metrics = None self.metrics = None