diff --git a/mu_map/training/cgan.py b/mu_map/training/cgan.py index c9a6b9b5e190d36d8d320f89a1683eee2bbc76bf..9a69d296735626fc823e3862dd9658484cbcc564 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,