Ethan Rosenthal

Thanks @hanlint ! @antoinebrl Yup, I did the same thing! Relatedly, I also had to convert `outputs` to full precision within `update_metric` if I tried to train with mixed precision.