Removing wraps from costs/gated_autoencoder.py