def batch_size(self, epoch, max_size):
    exponent = (epoch // 100)
    return min(max((2 ** exponent), 2), max_size)
def batch_size(self, epoch, max_size):
    exponent = (epoch // 100)
    return min(max((2 ** exponent), 2), max_size)