2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Added missing argument

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-04-29 11:49:58 +02:00
parent 55481402a8
commit 3ee6f3c780
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/twomartens/masterthesis/main.py
View File

@ -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: