Method save

axlearn/common/checkpointer_test.py:1427–1429

()

                step = 1000
                # Save ckpt.
                final_dir = os.path.join(root_dir, f"step_{step:08d}")

                async def save():
                    storage.save_to_dir(step=step, state=state, ckpt_dir=final_dir)
                    storage.wait_until_finished()