Text generate arguments.
(parser)
| 246 | |
| 247 | |
| 248 | def add_text_generate_args(parser): |
| 249 | """Text generate arguments.""" |
| 250 | |
| 251 | group = parser.add_argument_group('Text generation', 'configurations') |
| 252 | group.add_argument("--temperature", type=float, default=1.0) |
| 253 | group.add_argument("--top-p", type=float, default=0.0) |
| 254 | group.add_argument("--top-k", type=int, default=0) |
| 255 | group.add_argument("--out-seq-length", type=int, default=256) |
| 256 | group.add_argument("--num-beams", type=int, default=1) |
| 257 | group.add_argument("--length-penalty", type=float, default=0.0) |
| 258 | group.add_argument("--no-repeat-ngram-size", type=int, default=0) |
| 259 | group.add_argument("--min-tgt-length", type=int, default=0) |
| 260 | group.add_argument("--select-topk", action='store_true') |
| 261 | group.add_argument("--blank-maskratio", type=float, default=0.1) |
| 262 | return parser |
| 263 | |
| 264 | |
| 265 | def add_data_args(parser): |