diff --git a/train_with_tensorboard.py b/train_with_tensorboard.py
index 8d14e2c1b8074406011c62b31b45b535e05a6827..93a3c2a1cf66d804afd6f8d64f7447b7b02261c5 100644
--- a/train_with_tensorboard.py
+++ b/train_with_tensorboard.py
@@ -272,7 +272,7 @@ if __name__ == "__main__":
             gen = Generator(Batch_size, lines[:num_train],
                             (input_shape[0], input_shape[1])).generate(train=True, mosaic = mosaic)
             gen_val = Generator(Batch_size, lines[num_train:],
-                            (input_shape[0], input_shape[1])).generate(train=False, mosaic = mosaic)
+                            (input_shape[0], input_shape[1])).generate(train=False, mosaic = False)
 
         epoch_size = max(1, num_train//Batch_size)
         epoch_size_val = num_val//Batch_size
@@ -309,7 +309,7 @@ if __name__ == "__main__":
             gen = Generator(Batch_size, lines[:num_train],
                             (input_shape[0], input_shape[1])).generate(train=True, mosaic = mosaic)
             gen_val = Generator(Batch_size, lines[num_train:],
-                            (input_shape[0], input_shape[1])).generate(train=False, mosaic = mosaic)
+                            (input_shape[0], input_shape[1])).generate(train=False, mosaic = False)
 
         epoch_size = max(1, num_train//Batch_size)
         epoch_size_val = num_val//Batch_size