For offline model evaluation, we need to checkpoint global and local models. However, currently only global models are checkpointed. We need to extend it to cover local models.