From c0e5f74beebdc8a2258a35bf6cc3084bb3bff296 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Mon, 3 Jun 2019 10:37:08 +0200 Subject: [PATCH] Reduced forward passes to 10 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 f1f2fa3..9c8e13f 100644 --- a/src/twomartens/masterthesis/main.py +++ b/src/twomartens/masterthesis/main.py @@ -290,7 +290,7 @@ def _ssd_val(args: argparse.Namespace) -> None: batch_size = 16 image_size = 300 - forward_passes_per_image = 42 + forward_passes_per_image = 10 use_dropout = False if args.network == "ssd" else True weights_file = f"{args.weights_path}/VGG_coco_SSD_300x300_iter_400000.h5"