🔀 [Merge] branch 'MODEL' into TEST

yolo/tools/solver.py

@@ -1,3 +1,4 @@
+from math import ceil
 from pathlib import Path
 
 from lightning import LightningModule
@@ -83,7 +84,7 @@ class TrainModel(ValidateModel):
         return self.train_loader
 
     def on_train_epoch_start(self):
-        self.trainer.optimizers[0].next_epoch((len(self.train_loader) + 1) // self.trainer.world_size)
+        self.trainer.optimizers[0].next_epoch(ceil(len(self.train_loader) / self.trainer.world_size))
 
     def training_step(self, batch, batch_idx):
         lr_dict = self.trainer.optimizers[0].next_batch()