Fix incorrect device selection when resuming training (#4301)

single_channel
Laughing 1 year ago committed by GitHub
parent 28893d9d2d
commit b73f9b13df
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

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

Loading…
Cancel
Save