From 87a900a975eb16024842b2ac6fa155edec9aba00 Mon Sep 17 00:00:00 2001 From: Erik Bernhardsson Date: Fri, 13 Apr 2018 12:39:17 -0400 Subject: [PATCH] lower batch size --- convoys/tf_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/convoys/tf_utils.py b/convoys/tf_utils.py index 376dd44..c33de2a 100644 --- a/convoys/tf_utils.py +++ b/convoys/tf_utils.py @@ -11,7 +11,7 @@ def get_batch_placeholders(vs): def optimize(sess, target_batch, target_global=None, placeholders={}, - batch_size=1024, update_callback=None): + batch_size=128, update_callback=None): if placeholders: n = int(list(placeholders.values())[0].shape[0]) indexes = list(range(n))