From 1ea9ff98d4bc1d64e7e5c7cdc5b38885ee3afc2d Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Fri, 12 Apr 2019 14:17:17 +0200 Subject: [PATCH] Increased zsize to 64 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 38c2067..65c4759 100644 --- a/src/twomartens/masterthesis/main.py +++ b/src/twomartens/masterthesis/main.py @@ -111,7 +111,7 @@ def _auto_encoder_train(args: argparse.Namespace) -> None: with train_summary_writer.as_default(): train.train_simple(coco_data, iteration=args.iteration, weights_prefix=f"{args.weights_path}/category-{category}", - zsize=8, lr=0.0001, + zsize=64, lr=0.0001, channels=3, train_epoch=args.num_epochs, batch_size=batch_size)