Moved update max norm from prediction to learn in gd_pistol.cc