HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

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
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