Solved bug at autoencoder training