in `child_modules` the forward pass example should use the encoder subnet
📚 Documentation
In my understanding, the forward pass of the example autoencoder should return the latent representation extracted by the encoder.
If my assumption is correct, then the forward, training_step, and _shared_eval functions should be corrected to reflect that. In this case, I'd be happy to create and submit a PR.
Otherwise, I am deeply confused and could use an explanation 😅
Thanks for the amazing library!