From 9acbcad90cf31fa3a0cf02b80cb406d0080ed2b0 Mon Sep 17 00:00:00 2001 From: Xudong Sun Date: Sat, 7 Oct 2023 16:33:35 +0200 Subject: [PATCH 1/2] Update args_vae.py --- domainlab/models/args_vae.py | 5 ----- 1 file changed, 5 deletions(-) diff --git a/domainlab/models/args_vae.py b/domainlab/models/args_vae.py index 5fd22aa69..7ac8ae9a0 100644 --- a/domainlab/models/args_vae.py +++ b/domainlab/models/args_vae.py @@ -46,9 +46,4 @@ def add_args2parser_vae(parser): help='multiplier for KL x') parser.add_argument('--beta_y', type=float, default=1., help='multiplier for KL y') - # - - parser.add_argument('-w', '--warmup', type=int, default=100, - help='number of epochs for hyper-parameter warm-up. \ - Set to 0 to turn warmup off.') return parser From 8f1441866313cd970389a0d53da5d87095f9cb75 Mon Sep 17 00:00:00 2001 From: Xudong Sun Date: Sat, 7 Oct 2023 16:34:33 +0200 Subject: [PATCH 2/2] Update arg_parser.py --- domainlab/arg_parser.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/domainlab/arg_parser.py b/domainlab/arg_parser.py index 53deb148a..4f79a3ed1 100644 --- a/domainlab/arg_parser.py +++ b/domainlab/arg_parser.py @@ -52,6 +52,10 @@ def mk_parser_main(): parser.add_argument('--epo_te', default=1, type=int, help='test performance per {} epochs') + + parser.add_argument('-w', '--warmup', type=int, default=100, + help='number of epochs for hyper-parameter warm-up. \ + Set to 0 to turn warmup off.') parser.add_argument('--debug', action='store_true', default=False) parser.add_argument('--dmem', action='store_true', default=False)