Reduced forward passes to 10

Signed-off-by: Jim Martens <github@2martens.de>
2019-06-03 10:37:08 +02:00
parent 145b929863
commit c0e5f74bee
1 changed files with 1 additions and 1 deletions
--- 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"