Fix Autoencoder yaml files