At the moment, when the autoencoders are instances of SPADEAutoencoder, the code fails. This case should be added.
At the moment, when the autoencoders are instances of SPADEAutoencoder, the code fails.
This case should be added.