style changes

neuralmagic · Jun 16, 2023 · 3652a29 · 3652a29
1 parent 1a59320
commit 3652a29
Showing 1 changed file with 7 additions and 2 deletions.
diff --git 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: