From 53b37031ab3bf2767f4fb43878a0e0443db51fa1 Mon Sep 17 00:00:00 2001 From: Abhinav Agarwalla Date: Fri, 21 Apr 2023 15:55:57 +0000 Subject: [PATCH 1/4] Reducing validation batch size --- src/sparseml/yolov8/trainers.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/sparseml/yolov8/trainers.py b/src/sparseml/yolov8/trainers.py index 2875f0b650e..56e24f18817 100644 --- a/src/sparseml/yolov8/trainers.py +++ b/src/sparseml/yolov8/trainers.py @@ -277,6 +277,9 @@ def _setup_train(self, rank, world_size): super()._setup_train(rank, world_size) # NOTE: self.resume_training() was called in ^ + if rank in {0, -1}: + self.test_loader = self.get_dataloader(self.testset, batch_size=self.train_loader.batch_size // 4, rank=-1, mode='val') + if rank in {0, -1}: config = dict(self.args) if self.manager is not None: From 31445ac1390a30a186c42b39985f732e903e5646 Mon Sep 17 00:00:00 2001 From: Abhinav Agarwalla Date: Fri, 21 Apr 2023 16:36:05 +0000 Subject: [PATCH 2/4] updating validator --- src/sparseml/yolov8/trainers.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/sparseml/yolov8/trainers.py b/src/sparseml/yolov8/trainers.py index 56e24f18817..d359ceb717c 100644 --- a/src/sparseml/yolov8/trainers.py +++ b/src/sparseml/yolov8/trainers.py @@ -279,7 +279,8 @@ def _setup_train(self, rank, world_size): if rank in {0, -1}: self.test_loader = self.get_dataloader(self.testset, batch_size=self.train_loader.batch_size // 4, rank=-1, mode='val') - + self.validator = self.get_validator() + if rank in {0, -1}: config = dict(self.args) if self.manager is not None: From 1a59320588a31ab109cdd284a6dc0e677aa50c2a Mon Sep 17 00:00:00 2001 From: Abhinav Agarwalla Date: Thu, 15 Jun 2023 22:36:45 +0000 Subject: [PATCH 3/4] set minimum to 1 --- src/sparseml/yolov8/trainers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/sparseml/yolov8/trainers.py b/src/sparseml/yolov8/trainers.py index c63f90077b9..0d468621995 100644 --- a/src/sparseml/yolov8/trainers.py +++ b/src/sparseml/yolov8/trainers.py @@ -280,7 +280,7 @@ def _setup_train(self, rank, world_size): # NOTE: self.resume_training() was called in ^ if rank in {0, -1}: - self.test_loader = self.get_dataloader(self.testset, batch_size=self.train_loader.batch_size // 4, rank=-1, mode='val') + self.test_loader = self.get_dataloader(self.testset, batch_size=max(1, self.train_loader.batch_size // 4), rank=-1, mode='val') self.validator = self.get_validator() if rank in {0, -1}: From 3652a29ccc586423d19b3f5ded7a94057cfd305c Mon Sep 17 00:00:00 2001 From: Abhinav Agarwalla Date: Fri, 16 Jun 2023 07:56:59 +0000 Subject: [PATCH 4/4] style changes --- src/sparseml/yolov8/trainers.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/src/sparseml/yolov8/trainers.py b/src/sparseml/yolov8/trainers.py index 0d468621995..2b3d3d299a5 100644 --- a/src/sparseml/yolov8/trainers.py +++ b/src/sparseml/yolov8/trainers.py @@ -280,9 +280,14 @@ def _setup_train(self, rank, world_size): # NOTE: self.resume_training() was called in ^ if rank in {0, -1}: - self.test_loader = self.get_dataloader(self.testset, batch_size=max(1, self.train_loader.batch_size // 4), rank=-1, mode='val') + self.test_loader = self.get_dataloader( + self.testset, + batch_size=max(1, self.train_loader.batch_size // 4), + rank=-1, + mode="val", + ) self.validator = self.get_validator() - + if rank in {0, -1}: config = dict(self.args) if self.manager is not None: