Text generate arguments.
(parser)
| 267 | |
| 268 | |
| 269 | def add_text_generate_args(parser): |
| 270 | """Text generate arguments.""" |
| 271 | |
| 272 | group = parser.add_argument_group('Text generation', 'configurations') |
| 273 | group.add_argument("--temperature", type=float, default=1.0) |
| 274 | group.add_argument("--top_p", type=float, default=0.0) |
| 275 | group.add_argument("--top_k", type=int, default=0) |
| 276 | group.add_argument("--num-beams", type=int, default=1) |
| 277 | group.add_argument("--length-penalty", type=float, default=0.0) |
| 278 | group.add_argument("--no-repeat-ngram-size", type=int, default=0) |
| 279 | group.add_argument("--min-tgt-length", type=int, default=0) |
| 280 | group.add_argument("--out-seq-length", type=int, default=256) |
| 281 | group.add_argument('--input-source', type=str, default='interactive', |
| 282 | help='what input mode to use, interactive or path') |
| 283 | group.add_argument('--output-path', type=str, default='./samples', |
| 284 | help='path to place the generated samples') |
| 285 | group.add_argument('--with-id', action='store_true', |
| 286 | help='If each line is prepended with an id.') |
| 287 | group.add_argument('--max-inference-batch-size', type=int, default=12) |
| 288 | group.add_argument('--device', type=int, default=-1) |
| 289 | return parser |
| 290 | |
| 291 | |
| 292 |