Update arg_parser.py

marrlab · Oct 7, 2023 · 8f14418 · 8f14418
1 parent 9acbcad
commit 8f14418
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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)