rename METRIC_SUB_BATCH_SIZE to METRIC_BATCH_SIZE to match actual purpose