From 7249d569f75a2c68073e1209bc3d30869796a545 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Wed, 17 Apr 2019 14:23:51 +0200 Subject: [PATCH] Fixed zsize for val 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 5c902bf..6b90960 100644 --- a/src/twomartens/masterthesis/main.py +++ b/src/twomartens/masterthesis/main.py @@ -132,11 +132,11 @@ def _val(args: argparse.Namespace) -> None: with use_summary_writer.as_default(): run.run_simple(coco_data, iteration=args.iteration_trained, weights_prefix=f"{args.weights_path}/category-{category_trained}", - zsize=64, verbose=args.verbose, channels=3, batch_size=batch_size) + zsize=32, verbose=args.verbose, channels=3, batch_size=batch_size) else: run.run_simple(coco_data, iteration=args.iteration_trained, weights_prefix=f"{args.weights_path}/category-{category_trained}", - zsize=64, verbose=args.verbose, channels=3, batch_size=batch_size) + zsize=32, verbose=args.verbose, channels=3, batch_size=batch_size) def _auto_encoder_train(args: argparse.Namespace) -> None: