From 3439693c25364457eb31fe5b5fcee0bbe46ac00a Mon Sep 17 00:00:00 2001
From: Dennis Noll <dennis.noll@rwth-aachen.de>
Date: Wed, 2 Sep 2020 10:16:17 +0200
Subject: [PATCH] [keras] PlotCallback: raise evaluation batch size
---
keras.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/keras.py b/keras.py
index d25c563..1911c2c 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 = {}
--
GitLab