Merge branch 'master' into fbopt

marrlab · Oct 8, 2023 · 6a787a2 · 6a787a2
2 parents 358c808 + e7aae83
commit 6a787a2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/domainlab/arg_parser.py b/domainlab/arg_parser.py
@@ -53,6 +53,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)

diff --git 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