Learning scheduler

Hi Patrick

I am using latest version of pytorch.