joechou commited on
Commit
8e7f431
·
1 Parent(s): 31d8b31

Update options.py

Browse files
Files changed (1) hide show
  1. options.py +1 -1
options.py CHANGED
@@ -20,7 +20,7 @@ def add_options(parser: ArgumentParser):
20
  parser.add_argument("--num-encoder-layers", default=6, type=int)
21
  parser.add_argument("--num-decoder-layers", default=6, type=int)
22
  parser.add_argument("--max-seq-len", default=256, type=int)
23
- parser.add_argument("--mode", default="baseline", type=str, choices=[
24
  helpers.BASELINE, helpers.KOBE_ATTRIBUTE, helpers.KOBE_KNOWLEDGE, helpers.KOBE_FULL])
25
 
26
  # Training
 
20
  parser.add_argument("--num-encoder-layers", default=6, type=int)
21
  parser.add_argument("--num-decoder-layers", default=6, type=int)
22
  parser.add_argument("--max-seq-len", default=256, type=int)
23
+ parser.add_argument("--mode", default="kobe-full", type=str, choices=[
24
  helpers.BASELINE, helpers.KOBE_ATTRIBUTE, helpers.KOBE_KNOWLEDGE, helpers.KOBE_FULL])
25
 
26
  # Training