Update sparse_autoencoder.py