From 3ee6f3c7807d6f36f9dec6774414866d5dd98e90 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Mon, 29 Apr 2019 11:49:58 +0200 Subject: [PATCH] Added missing argument Signed-off-by: Jim Martens --- src/twomartens/masterthesis/main.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/twomartens/masterthesis/main.py b/src/twomartens/masterthesis/main.py index dc24bcd..d1aa41f 100644 --- a/src/twomartens/masterthesis/main.py +++ b/src/twomartens/masterthesis/main.py @@ -195,9 +195,9 @@ def _ssd_val(args: argparse.Namespace) -> None: ) if args.debug: with use_summary_writer.as_default(): - ssd.predict(scenenet_data, use_dropout, output_path, weights_file) + ssd.predict(scenenet_data, use_dropout, output_path, weights_file, nr_digits=nr_digits) else: - ssd.predict(scenenet_data, use_dropout, output_path, weights_file) + ssd.predict(scenenet_data, use_dropout, output_path, weights_file, nr_digits=nr_digits) def _auto_encoder_val(args: argparse.Namespace) -> None: