Spaces:
Runtime error
Runtime error
Update options.py
Browse files- 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="
|
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
|