start at lr = 0.01 again as it no longer diverges with bugs etc fixed