Added chunk_size into the optimizer_params