Child Modules Code Example
📚 Documentation
For typos and doc fixes, please go ahead and:

Create an issue.
My issues are all about the documentation page about child modules: https://pytorch-lightning.readthedocs.io/en/latest/child_modules.html#child-modules

a) The code example refers to a variable named logits in the method _shared_eval(..).
What is logits referring to?
b) forward returns the result of calling decoder. This is correctly used in training_step. But _shared_eval seems to expect a classifier output when calling self(representation).
c) Even more confusing in the third code block, calling the autoencoder returns the representation:
some_images = torch.Tensor(32, 1, 28, 28)
representations = autoencoder(some_images)

So in

training_step -> decoding
_shared_eval -> classifier
inference -> encoding

Or is there some underlying logic which switches between the implementations?
d) Probably connected with the other issues: The forward-function of LitMNIST, which implements the classficiation, required in _shared_eval, is overriden, but not accessed.

Fix the typo.
No typo.

Not sure what the correct way to rewrite the example is.
I can fix the issues if someone answers my questions/explains the issues I presented in 1.

Submit a PR.
Will do, when I know how to fix the issues.