Schedule learning rate decay

The problem was related to this. Enjoy the rest of the course, @cpedrosa!

P.S.: Good hint, @AnkitSaini :slight_smile: