https://github.com/Overworldai/owl-vaes/blob/32d3b84911490babeed41a4b369ee4cbdaa20199/owl_vaes/models/dcae.py#L212 When model is loaded in eval() mode, encoder returns only mu, but DCAE forward expects mu, logvar