Fix resume (#138)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-01-03 21:06:22 +08:00 · 2023-01-03 21:06:22 +08:00 · 340376f7a6
commit 340376f7a6
parent 82c849c163
9 changed files with 55 additions and 26 deletions
--- a/ultralytics/yolo/engine/validator.py
+++ b/ultralytics/yolo/engine/validator.py
@ -111,6 +111,7 @@ class BaseValidator:
                self.args.workers = 0  # faster CPU val as time dominated by inference, not dataloading
            self.dataloader = self.dataloader or \
                              self.get_dataloader(data.get("val") or data.set("test"), self.args.batch_size)
+            self.data = data

        model.eval()