From 8678f14083713178667552c4df91c2462d5f4f0d Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Thu, 20 Jun 2019 14:08:20 +0200 Subject: [PATCH] Update correct batch size Signed-off-by: Jim Martens --- src/twomartens/masterthesis/cli.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/twomartens/masterthesis/cli.py b/src/twomartens/masterthesis/cli.py index db84fc3..0fd2a59 100644 --- a/src/twomartens/masterthesis/cli.py +++ b/src/twomartens/masterthesis/cli.py @@ -46,7 +46,7 @@ def _ssd_train(args: argparse.Namespace) -> None: tf.enable_eager_execution() - batch_size = 16 + batch_size = 32 image_size = 300 use_dropout = False if args.network == "ssd" else True @@ -118,7 +118,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 image_size = 256 coco_data = data.load_coco_train(coco_path, category, num_epochs=args.num_epochs, batch_size=batch_size, resized_shape=(image_size, image_size))