BATCH_SIZE_BY_CONTEXT = {
    32: 16,
    64: 8,
    128: 4,
    256: 2,
}

        __
        __