Skip to content
Snippets Groups Projects
Commit 3439693c authored by Dennis Noll's avatar Dennis Noll
Browse files

[keras] PlotCallback: raise evaluation batch size

parent f9ce0c99
No related branches found
No related tags found
No related merge requests found
......@@ -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 = {}
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment