diff --git a/keras.py b/keras.py
index d25c563bfe12917887475a861996fd09429b034c..1911c2c92559a7506d4c455c6bc3ad8afc9b83cd 100644
--- a/keras.py
+++ b/keras.py
@@ -467,7 +467,7 @@ class PlotMulticlass(TFSummaryCallback):
         self.make_plots(epoch, logs)
 
     def make_plots(self, epoch, logs):
-        prediction = self.model.predict(self.x)
+        prediction = self.model.predict(self.x, batch_size=4096)
         truth = self.truth
         imgs = {}