Set None batch_id on_batch_end