diff --git a/ultralytics/engine/model.py b/ultralytics/engine/model.py index db8d87eb..8affd958 100644 --- a/ultralytics/engine/model.py +++ b/ultralytics/engine/model.py @@ -115,7 +115,7 @@ class Model(nn.Module): self.predictor = None # reuse predictor self.model = None # model object self.trainer = None # trainer object - self.ckpt = None # if loaded from *.pt + self.ckpt = {} # if loaded from *.pt self.cfg = None # if loaded from *.yaml self.ckpt_path = None self.overrides = {} # overrides for trainer object @@ -807,7 +807,7 @@ class Model(nn.Module): # Update model and cfg after training if RANK in {-1, 0}: ckpt = self.trainer.best if self.trainer.best.exists() else self.trainer.last - self.model, _ = attempt_load_one_weight(ckpt) + self.model, self.ckpt = attempt_load_one_weight(ckpt) self.overrides = self.model.args self.metrics = getattr(self.trainer.validator, "metrics", None) # TODO: no metrics returned by DDP return self.metrics