Add checkpointing guide and configurations to PyTorch TPU recipes

AI-Hypercomputer / tpu-recipes

Apache License 2.0

2 stars 0 forks source link

Add checkpointing guide and configurations to PyTorch TPU recipes #7

Open miladm opened 4 days ago

miladm commented 4 days ago

Add checkpointing guide and configurations to existing PyTorch TPU recipes. The guide should offer solutions for distributed checkpointing and consolidating checkpoints.