add learning_rate args for auto-encoder