Fixed typo in documentation of `LazyModuleMixin`
- What changed?
- On line `#92` there was a typo (`mlp`). Changed it to `lazy_mlp`.
```diff
- >>> lazy_mlp = mlp.cuda().double()
+ >>> lazy_mlp = lazy_mlp.cuda().double()
```
- Docs: [`LazyModuleMixin`](https://pytorch.org/docs/1.11/_modules/torch/nn/modules/lazy.html#LazyModuleMixin)
Closes #76269.
https://github.com/pytorch/pytorch/blob/1a7e43be141ce01469d7605075cb1008bf19abd7/torch/nn/modules/lazy.py#L89-L93
Pull Request resolved: https://github.com/pytorch/pytorch/pull/76269
Approved by: https://github.com/mrshenli