Use the same shuffling/load balancing parameters as the checkpointed model when
restarting.