Checkpointing at every model tuning stage

Hello,

Is there a way we can checkpoint and save the model at every stage in the AutoML training process? If not, where in the code can we modify it?

Thanks.