Loading nn.Module from checkpoint tutorial (non ghstack)

[mikaylagawarecki]

Added tutorial for some utilities for loading checkpoint

1. The mmap keyword argument on torch.load
2. The torch.device() context manager
3. The assign keyword argument on nn.Module.load_state_dict()

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/tutorials/2522

• 📄 Preview Python docs built from this PR

Note: Links to docs will display an error until the docs builds have been completed.

✅ No Failures

As of commit 903b8c4 with merge base 646c8b6 ():
💚 Looks good so far! There are no failures yet. 💚

This comment was automatically generated by Dr. CI and updates every 15 minutes.

[svekars]

thanks, @mikaylagawarecki. Looks good, just a couple of editorial suggestions.

[albanD]

SGTM