adding code to train an autoencoder on chroma