Fix learning rate adjustment.

Try using page_locked memory.