Added learning rate scaling. Thanks @marcofraccaro