@@ -35,12 +35,12 @@ def main(): | |||||
"--batch-size", type=int, default=32, help="base model batch size (default : 32)" | "--batch-size", type=int, default=32, help="base model batch size (default : 32)" | ||||
) | ) | ||||
parser.add_argument( | parser.add_argument( | ||||
"--segment_size", type=int or float, default=1000, help="segment size (default : 1000)" | |||||
"--segment_size", type=int, default=1000, help="segment size (default : 1000)" | |||||
) | ) | ||||
parser.add_argument("--save_interval", type=int, default=1, help="save interval (default : 1)") | parser.add_argument("--save_interval", type=int, default=1, help="save interval (default : 1)") | ||||
parser.add_argument( | parser.add_argument( | ||||
"--max-revision", | "--max-revision", | ||||
type=int or float, | |||||
type=int, | |||||
default=10, | default=10, | ||||
help="maximum revision in reasoner (default : 10)", | help="maximum revision in reasoner (default : 10)", | ||||
) | ) | ||||
@@ -87,19 +87,19 @@ def main(): | |||||
"--loops", type=int, default=5, help="number of loop iterations (default : 5)" | "--loops", type=int, default=5, help="number of loop iterations (default : 5)" | ||||
) | ) | ||||
parser.add_argument( | parser.add_argument( | ||||
"--segment_size", type=int or float, default=1000, help="segment size (default : 1000)" | |||||
"--segment_size", type=int, default=1000, help="segment size (default : 1000)" | |||||
) | ) | ||||
parser.add_argument("--save_interval", type=int, default=1, help="save interval (default : 1)") | parser.add_argument("--save_interval", type=int, default=1, help="save interval (default : 1)") | ||||
parser.add_argument( | parser.add_argument( | ||||
"--max-revision", | "--max-revision", | ||||
type=int or float, | |||||
type=int, | |||||
default=-1, | default=-1, | ||||
help="maximum revision in reasoner (default : -1)", | help="maximum revision in reasoner (default : -1)", | ||||
) | ) | ||||
parser.add_argument( | parser.add_argument( | ||||
"--require-more-revision", | "--require-more-revision", | ||||
type=int, | type=int, | ||||
default=5, | |||||
default=0, | |||||
help="require more revision in reasoner (default : 0)", | help="require more revision in reasoner (default : 0)", | ||||
) | ) | ||||
parser.add_argument( | parser.add_argument( | ||||
@@ -53,12 +53,12 @@ def main(): | |||||
"--loops", type=int, default=2, help="number of loop iterations (default : 2)" | "--loops", type=int, default=2, help="number of loop iterations (default : 2)" | ||||
) | ) | ||||
parser.add_argument( | parser.add_argument( | ||||
"--segment_size", type=int or float, default=0.01, help="segment size (default : 0.01)" | |||||
"--segment_size", type=int, default=0.01, help="segment size (default : 0.01)" | |||||
) | ) | ||||
parser.add_argument("--save_interval", type=int, default=1, help="save interval (default : 1)") | parser.add_argument("--save_interval", type=int, default=1, help="save interval (default : 1)") | ||||
parser.add_argument( | parser.add_argument( | ||||
"--max-revision", | "--max-revision", | ||||
type=int or float, | |||||
type=int, | |||||
default=-1, | default=-1, | ||||
help="maximum revision in reasoner (default : -1)", | help="maximum revision in reasoner (default : -1)", | ||||
) | ) | ||||