From 37b0b702a3306aee6e63de74082ba5968d59caf2 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Fri, 12 Apr 2019 13:42:48 +0200 Subject: [PATCH] Changed batch size to 16 Signed-off-by: Jim Martens --- src/twomartens/masterthesis/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/twomartens/masterthesis/main.py b/src/twomartens/masterthesis/main.py index b2eed02..2bbcff7 100644 --- a/src/twomartens/masterthesis/main.py +++ b/src/twomartens/masterthesis/main.py @@ -102,7 +102,7 @@ def _auto_encoder_train(args: argparse.Namespace) -> None: tf.enable_eager_execution() coco_path = args.coco_path category = args.category - batch_size = 32 + batch_size = 16 coco_data = data.load_coco(coco_path, category, num_epochs=args.num_epochs, batch_size=batch_size, resized_shape=(64, 64)) train_summary_writer = summary_ops_v2.create_file_writer(