Evaluation arguments.
(parser)
| 186 | |
| 187 | |
| 188 | def add_evaluation_args(parser): |
| 189 | """Evaluation arguments.""" |
| 190 | |
| 191 | group = parser.add_argument_group('validation', 'validation configurations') |
| 192 | |
| 193 | group.add_argument('--eval-batch-size', type=int, default=None, |
| 194 | help='Data Loader batch size for evaluation datasets.' |
| 195 | 'Defaults to `--batch-size`') |
| 196 | group.add_argument('--eval-iters', type=int, default=100, |
| 197 | help='number of iterations to run for evaluation' |
| 198 | 'validation/test for') |
| 199 | group.add_argument('--eval-interval', type=int, default=1000, |
| 200 | help='interval between running evaluation on validation set') |
| 201 | |
| 202 | return parser |
| 203 | |
| 204 | |
| 205 | def add_text_generate_args(parser): |