Change autoencoder so that it uses square input data (allows full convolution to bring input back to size). Should work now