Text generate arguments.
(parser)
| 207 | return parser |
| 208 | |
| 209 | def add_text_generate_args(parser): |
| 210 | """Text generate arguments.""" |
| 211 | |
| 212 | group = parser.add_argument_group('Text generation', 'configurations') |
| 213 | group.add_argument("--temperature", type=float, default=1.0) |
| 214 | group.add_argument("--top_p", type=float, default=0.0) |
| 215 | group.add_argument("--top_k", type=int, default=0) |
| 216 | group.add_argument("--out-seq-length", type=int, default=256) |
| 217 | return parser |
| 218 | |
| 219 | |
| 220 | def add_data_args(parser): |