Fix incorrect device selection when resuming training (#4301)

This commit is contained in:
Laughing
2023-08-11 20:32:39 +08:00
committed by GitHub
parent 28893d9d2d
commit b73f9b13df

View File

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