How to save model state in pytorch fsdp

I am using PyTorch’s FSDP. How can I save the completed model state every 100 batches?
Is the following code correct?