Issue with Variational encoder - while doing forward pass with Pytorch lightning

vision
1
  • I followed @Kinyugo - notebook at
    Stackoverflow - Kinyugo
    and the variational_encoder python script and even with the refactored LitVAE code, I am getting error at the forward pass.

Attaching the screenshot enter image description here

@ptrblck_de - Any help is greatly appreciated. TIA.