Fix incorrect device selection when resuming training (#4301)

2 years ago · b73f9b13df
parent 28893d9d2d
commit b73f9b13df
1 changed files with 1 additions and 1 deletions
--- a/ultralytics/engine/trainer.py
+++ b/ultralytics/engine/trainer.py
@ -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