plot all the images

3 months ago · af4d806be6
parent 26298592d1
commit af4d806be6
1 changed files with 1 additions and 1 deletions
--- a/ultralytics/engine/trainer.py
+++ b/ultralytics/engine/trainer.py
@ -418,7 +418,7 @@ class BaseTrainer:
                        )
                    )
                    self.run_callbacks("on_batch_end")
-                    if self.args.plots and ni in self.plot_idx:
+                    if self.args.plots:
                        self.plot_training_samples(batch, ni)
                self.run_callbacks("on_train_batch_end")