Evaluation arguments.
(parser)
| 207 | |
| 208 | |
| 209 | def add_evaluation_args(parser): |
| 210 | """Evaluation arguments.""" |
| 211 | |
| 212 | group = parser.add_argument_group('validation', 'validation configurations') |
| 213 | |
| 214 | group.add_argument('--eval-batch-size', type=int, default=None, |
| 215 | help='Data Loader batch size for evaluation datasets.' |
| 216 | 'Defaults to `--batch-size`') |
| 217 | group.add_argument('--eval-iters', type=int, default=100, |
| 218 | help='number of iterations to run for evaluation' |
| 219 | 'validation/test for') |
| 220 | group.add_argument('--eval-interval', type=int, default=None, |
| 221 | help='interval between running evaluation on validation set') |
| 222 | group.add_argument('--strict-eval', action='store_true', |
| 223 | help='won\'t enlarge or randomly map eval-data, and eval full eval-data.') |
| 224 | return parser |
| 225 | |
| 226 | |
| 227 | def add_data_args(parser): |