From 90df91cb82e5eaaeb2cb0df6ef4270cf52ad2eef Mon Sep 17 00:00:00 2001
From: Tamino Huxohl <thuxohl@techfak.uni-bielefeld.de>
Date: Thu, 6 Oct 2022 15:03:38 +0200
Subject: [PATCH] update gan training loss weights
---
mu_map/training/cgan.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/mu_map/training/cgan.py b/mu_map/training/cgan.py
index c9a6b9b..9a69d29 100644
--- a/mu_map/training/cgan.py
+++ b/mu_map/training/cgan.py
@@ -434,9 +434,9 @@ if __name__ == "__main__":
lr_g=0.001,
lr_decay_factor_g=0.99,
lr_decay_epoch_g=1,
- l2_weight=1.0,
- gdl_weight=1.0,
- adv_weight=20.0,
+ l2_weight=0.25,
+ gdl_weight=0.25,
+ adv_weight=0.5,
snapshot_dir=args.snapshot_dir,
snapshot_epoch=args.snapshot_epoch,
logger=logger,
--
GitLab